bIg-RddlmZgdZddlZddlZddlmZmZmZddlm 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 eDfZEdZFddZGddZHddZI d d!dZJ d" d#dZK d d$dZLeFf d%dZMeFf d&dZN d dddddd d'dZO d d(dZPeKZQeMZReOZSy))) annotations)8ONNX_ML IR_VERSIONIR_VERSION_2017_10_10IR_VERSION_2017_10_30IR_VERSION_2017_11_3IR_VERSION_2019_1_22IR_VERSION_2019_3_18IR_VERSION_2019_9_19IR_VERSION_2020_5_8IR_VERSION_2021_7_30IR_VERSION_2023_5_5 EXPERIMENTALSTABLEcheckercomposedefs gen_protohelperhubmapping numpy_helperparserprintershape_inferenceutilsversion_converterAttributeProto FunctionProto GraphProtoMapProto ModelProto NodeProto OperatorProtoOperatorSetIdProtoOperatorSetProtoOperatorStatus OptionalProto SequenceProtoSparseTensorProtoStringStringEntryProtoTensorAnnotation TensorProtoTensorShapeProtoTrainingInfoProto TypeProtoValueInfoProtoVersionconvert_model_to_external_dataload_external_data_for_modelload_model_from_string load_modelload_tensor_from_string load_tensor save_model save_tensorwrite_external_data_tensorsN)IOLiteralUnion) serialization)r)r4r;r3)rrrr rrrrr r r r r rr"r#r%r'rr*r+r,r-r.r/r0r1r2)r$r&)r!r(r))version) rrrrrrrrrrrrr)protobuf textprotoonnxtxtjsonrAct|drettjtt |j r1tjtt |j }|Stjtttjf|}t|d5}|j }ddd|S#1swYSxYw)Nreadrb) hasattrcallabletypingcastr<bytesrFr>strosPathLikeopen)fcontentreadables J/var/www/html/answerous/venv/lib/python3.12/site-packages/onnx/__init__.py _load_bytesrUsq&hv{{2e9a'@'E'EF++bi+002 N KKc2;;./ 3 !T] &hmmoG & N & Ns /C  Cct|drettjtt |j r1tjtt |j |ytjtttjf|}t|d5}|j |dddy#1swYyxYw)Nwritewb) rHrIrJrKr<rLrWr>rMrNrOrP)rRrQwritables rT _save_bytesrZsq'x BuIq(A(G(GH BuIq!''0 KKc2;;./ 3 !T] $h NN7 # $ $ $s /C  Cct|ttjfrtjj |St |dr-|Jtjj |jSy)Nname) isinstancerMrNrOpathabspathrHr\)rQs rT_get_file_pathr`sV!c2;;'(wwq!!q&}}wwqvv&& c:|tjj|St|x}Atj j |\}}tjj|}|xst}|Jtjj|S)zQGet the serializer for the given path and format from the serialization registry.) r?registrygetr`rNr^splitextget_format_from_file_extension_DEFAULT_FORMAT)fmtrQ file_path_exts rT_get_serializerrls  %%))#..#A&& 3!!),3$$CCCH  C ??  ! ! % %c **raTct||jt|t}|r8t |}|r+t j 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. ) rldeserialize_protorUr"r`rNr^dirnamer4)rQformatload_external_datamodelmodel_filepathbase_dirs rTr6r6sT* FA & 8 8Q VE'* ww~6H ( 9 Lrac\t||jt|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. )rlrnrUr-)rQrps rTr8r8s$ 61 % 7 7 A VVracHt|j|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. )rlrnr"srps rTr5r5s 6 " 4 4Q EEracHt|j|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. )rlrnr-rws rTr7r7s 6 " 4 4Q FFraFi)save_as_external_dataall_tensors_to_one_filelocationsize_thresholdconvert_attributecRt|tr'ttj |t }|rt |||||t|}|+tjj|} t|| }t||j|} t| |y)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]rLrlrgrnr"r3r`rNr^ror;serialize_protorZ) protorQrprzr{r|r}r~rsbasepath serializeds rTr9r9sH%0BB5*,W& *HnFW $A&N!77??>2+E8< 8HHOJ AracRt||j|}t||y)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)rlrrZ)rrQrprs rTr:r:Ms% !+;;EBJ Ara)rQIO[bytes] | str | os.PathLikereturnrL)rRrLrQrrNone)rQz$IO[bytes] | str | os.PathLike | Noner str | None)N)rh_SupportedFormat | NonerQz$str | os.PathLike | IO[bytes] | Nonerzserialization.ProtoSerializer)NT)rQrrprrqboolrr")rQrrprrr-)rxz bytes | strrp_SupportedFormatrr")rxrLrprrr-)rzModelProto | bytesrQrrprrzrr{rr|rr}intr~rrr)rr-rQrrprrr)T __future__r__all__rNrJr<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__rrrrrrrrrrrrrrMrrgrUrZr`rlr6r8r5r7r9r:loadload_from_stringsaverarTrs3 #= @ %%+ <CDD/& 67<$MQ+ +%I+"+*'+#$ # D'+W$W #WW* /FF FF* /G G GG,'+2 #($(#2 2$2 $2  2 " 2222 2p'+ $ $ *)ra