Ng-ddlmZgdZddlZddlZddlmZmZmZddlm Z ddl m Z ddl m Z mZmZddlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,dd l-m.Z.m/Z/dd l0m1Z1m2Z2m3Z3dd l4m5Z6dd lm7Z7m8Z8m9Z9m:Z:m;Z;mZ>m?Z?m@Z@mAZAmBZBmCZCeed eDfZEdZFd9dZGd:dZHd;dZI dd?d$ZK dstrosPathLikeopen)rCcontentreadables I/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnx/__init__.py _load_bytesrUsq&&hv{2e9a'@'@'EFF&+bi++0022 Kc2;./ 3 3 !T]] &hmmooG & & & & & & & & & & & & & & & NsC""C&)C&rRNonect|drqttjtt |jr:tjtt ||dStjtttj f|}t|d5}||ddddS#1swxYwYdS)Nwritewb) rJrKrLrMr<rFrXr>rNrOrPrQ)rRrCwritables rT _save_bytesr[sq'$x BuIq(A(A(GHH$ BuIq!!''00000 Kc2;./ 3 3 !T]] $h NN7 # # # $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $sC&&C*-C*$IO[bytes] | str | os.PathLike | None str | Nonect|ttjfrtj|St |dr(|Jtj|jSdS)Nname) isinstancerNrOrPpathabspathrJr_)rCs rT_get_file_pathrcse!c2;'(("wq!!!q&'}}}wqv&&& 4fmt_SupportedFormat | None$str | os.PathLike | IO[bytes] | Noneserialization.ProtoSerializerc@|tj|St|x}Atj|\}}tj|}|pt}|Jtj|S)zQGet the serializer for the given path and format from the serialization registry.) r?registrygetrcrOrasplitextget_format_from_file_extension_DEFAULT_FORMAT)rerC file_path_exts rT_get_serializerrrs  %))#...#A&&& 3!!),,3$CCCHH  C ???  ! % %c * **rdTformatload_external_databoolr"ct||t|t}|r@t |}|r/t j|}t|||S)aLoads a serialized ModelProto into memory. Args: f: can be a file-like object (has "read" function) or a string/PathLike containing a file name format: The serialization format. When it is not specified, it is inferred from the file extension when ``f`` is a path. If not specified _and_ ``f`` is not a path, 'protobuf' is used. The encoding is assumed to be "utf-8" when the format is a text format. load_external_data: Whether to load the external data. Set to True if the data is under the same directory of the model. If not, users need to call :func:`load_external_data_for_model` with directory to load external data from. Returns: Loaded in-memory ModelProto. ) rrdeserialize_protorUr"rcrOradirnamer4)rCrsrtmodelmodel_filepathbase_dirs rTr6r6sr* FA & & 8 8Q V VE:'**  :w~66H ( 9 9 9 Lrdr-c|t||t|tS)aLoads a serialized TensorProto into memory. Args: f: can be a file-like object (has "read" function) or a string/PathLike containing a file name format: The serialization format. When it is not specified, it is inferred from the file extension when ``f`` is a path. If not specified _and_ ``f`` is not a path, 'protobuf' is used. The encoding is assumed to be "utf-8" when the format is a text format. Returns: Loaded in-memory TensorProto. )rrrwrUr-)rCrss rTr8r8s. 61 % % 7 7 A V VVrds bytes | str_SupportedFormatc`t||tS)aLoads a binary string (bytes) that contains serialized ModelProto. Args: s: a string, which contains serialized ModelProto format: The serialization format. When it is not specified, it is inferred from the file extension when ``f`` is a path. If not specified _and_ ``f`` is not a path, 'protobuf' is used. The encoding is assumed to be "utf-8" when the format is a text format. Returns: Loaded in-memory ModelProto. )rrrwr"r}rss rTr5r5s& 6 " " 4 4Q E EErdc`t||tS)aLoads a binary string (bytes) that contains serialized TensorProto. Args: s: a string, which contains serialized TensorProto format: The serialization format. When it is not specified, it is inferred from the file extension when ``f`` is a path. If not specified _and_ ``f`` is not a path, 'protobuf' is used. The encoding is assumed to be "utf-8" when the format is a text format. Returns: Loaded in-memory TensorProto. )rrrwr-rs rTr7r7s& 6 " " 4 4Q F FFrdFi)save_as_external_dataall_tensors_to_one_filelocationsize_thresholdconvert_attributeprotoModelProto | bytesrrrrintrct|tr4tt|t }|rt |||||t|}|/tj |} t|| }t|| |} t| |dS)a}Saves the ModelProto to the specified path and optionally, serialize tensors with raw data as external data before saving. Args: proto: should be a in-memory ModelProto f: can be a file-like object (has "write" function) or a string containing a file name or a pathlike object format: The serialization format. When it is not specified, it is inferred from the file extension when ``f`` is a path. If not specified _and_ ``f`` is not a path, 'protobuf' is used. The encoding is assumed to be "utf-8" when the format is a text format. save_as_external_data: If true, save tensors to external file(s). all_tensors_to_one_file: Effective only if save_as_external_data is True. If true, save all tensors to one external file specified by location. If false, save each tensor to a file named with the tensor name. location: Effective only if save_as_external_data is true. Specify the external file that all tensors to save to. Path is relative to the model path. If not specified, will use the model name. size_threshold: Effective only if save_as_external_data is True. Threshold for size of data. Only when tensor's data is >= the size_threshold it will be converted to external data. To convert every tensor with raw data to external data set size_threshold=0. convert_attribute: Effective only if save_as_external_data is True. If true, convert all tensors to external data If false, convert only non-attribute tensors to external data N)r`rFrrrnrwr"r3rcrOrarxr;serialize_protor[) rrCrsrrrrrrzbasepath serializeds rTr9r9sH%X00BB5*,,WW & *HnFW   $A&&N!7??>22+E8<< 88HHOOJ Ardclt|||}t||dS)aSaves the TensorProto to the specified path. Args: proto: should be a in-memory TensorProto f: can be a file-like object (has "write" function) or a string containing a file name or a pathlike object. format: The serialization format. When it is not specified, it is inferred from the file extension when ``f`` is a path. If not specified _and_ ``f`` is not a path, 'protobuf' is used. The encoding is assumed to be "utf-8" when the format is a text format. N)rrrr[)rrCrsrs rTr:r:Ks7 !++;;EBBJ Ard)rCrDrErF)rRrFrCrDrErV)rCr\rEr])N)rerfrCrgrErh)NT)rCrDrsrfrtrurEr")rCrDrsrfrEr-)r}r~rsrrEr")r}rFrsrrEr-)rrrCrDrsrfrrurrurr]rrrrurErV)rr-rCrDrsrfrErV)T __future__r__all__rOrLr<r=r>onnxr?onnx.onnx_cpp2py_exportronnx.external_data_helperr4r;r3 onnx.onnx_pbrrrr rrrrr r r r r rr"r#r%r'rr*r+r,r-r.r/r0r1r2onnx.onnx_operators_pbr$r&onnx.onnx_data_pbr!r(r) onnx.versionr@ __version__rrrrrrrrrrrrrrNrrnrUr[rcrrr6r8r5r7r9r:loadload_from_stringsaverdrTrs4 #"""""= = = @ %%%%%%%%%%++++++ <CBBBBBBBDDDDDDDDDD//////&!893>?$$$$MQ+++++*'+#D'+WWWWW* /FFFFF* /GGGGG,'+2 #($(#222222p'+*)rd