o soio;@s\ddlmZgdZddlZddlZddlmZmZddlmZddl m Z ddl m Z m Z mZddlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm 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,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2dd l3m4Z4m5Z5dd l6m7Z7m8Z8m9Z9ddl:Zdd lm;Z;mZ>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFejGrdd lHmIZIejJjJZKed eLBZMdZNdQddZOdRddZPdSddZQ dTdUd!d"ZR  #dVdWd(d)ZS dTdXd+d,ZTeNfdYd0d1ZUeNfdZd2d3ZV dTd4d#dd5d4d6d[d?d@ZW dTd\dAdBZXeSZYeUZZeWZ[d]dEdFZ\d^dHdIZ]d_dKdLZ^d`dOdPZ_e\e!_`e]e_`e^e_`dS)a) annotations)>ONNX_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_5IR_VERSION_2024_3_25 EXPERIMENTALSTABLEcheckercomposedefs gen_protohelperhub numpy_helperparserprintershape_inferenceutilsversion_converterAttributeProtoDeviceConfigurationProto FunctionProto GraphProtoIntIntListEntryProtoMapProto ModelProtoNodeDeviceConfigurationProto NodeProto OperatorProtoOperatorSetIdProtoOperatorSetProtoOperatorStatus OptionalProto SequenceProtoSimpleShardedDimProtoShardedDimProtoShardingSpecProtoSparseTensorProtoStringStringEntryProtoTensorAnnotation 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)IOLiteral) serialization)r)r9r@r8)#rrrrr r!rrrrrr r r r r rr#r$r%r'r)rr,r-r.r/r0r1r2r3r4r5r6r7)r&r()r"r*r+) rrrrrrrrrrrr)Sequence)protobuf textprotoonnxtxtjsonrEfIO[bytes] | str | os.PathLikereturnbytescCstt|drttd|jrtd|}|Std|}t|d }|}Wd|S1s3wY|S)Nread IO[bytes]str | os.PathLikerb)hasattrcallabletypingcastrMopen)rIcontentreadablerXA/home/ubuntu/.local/lib/python3.10/site-packages/onnx/__init__.py _load_bytess   rZrVNonecCsxt|drttd|jrtd||dStd|}t|d}||WddS1s5wYdS)NwriterNrOwb)rQrRrSrTr\rU)rVrIwritablerXrXrY _save_bytess    "r_$IO[bytes] | str | os.PathLike | None str | NonecCsDt|ttjfrtj|St|dr |dusJtj|jSdS)Nname) isinstancestrosPathLikepathabspathrQrb)rIrXrXrY_get_file_paths    rifmt_SupportedFormat | None$str | os.PathLike | IO[bytes] | Noneserialization.ProtoSerializercCs`|dur tj|St|}dur tj|\}}tj|}|p#t}|dus*Jtj|S)zQGet the serializer for the given path and format from the serialization registry.N) rCregistrygetrirergsplitextget_format_from_file_extension_DEFAULT_FORMAT)rjrI file_path_extrXrXrY_get_serializers    rvTformatload_external_databoolr#cCsBt||t|t}|rt|}|rtj|}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. ) rvdeserialize_protorZr#rirergdirnamer9)rIrwrxmodelmodel_filepathbase_dirrXrXrYr;s  r;r2cCst||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. )rvrzrZr2)rIrwrXrXrYr=sr=s bytes | str_SupportedFormatcCt||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. )rvrzr#rrwrXrXrYr:r:cCr)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. )rvrzr2rrXrXrYr<rr<Fi)save_as_external_dataall_tensors_to_one_filelocationsize_thresholdconvert_attributeprotoModelProto | bytesrrrrintrc Cstt|trtt|t}|rt|||||t|}|dur+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)rcrLrvrrrzr#r8rirergr{r@serialize_protor_) rrIrwrrrrrr}basepath serializedrXrXrYr>'s $   r>cCst|||}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)rvrr_)rrIrwrrXrXrYr?\sr?selfrdc Cs|jr d|jd}nd}|jrd|jd}nd}|jr%d|jd}nd}|jr1d|j}nd}|jr@dt|jd}nd}|jrNd t|j}nd}d |j||||||d S) N , domain=''z, producer_name='z, producer_version='z, graph=z , functions=, opset_import=zModelProto(ir_version=)) domain producer_nameproducer_versiongraph functionslen opset_import_operator_set_protos_repr ir_version)rrrrrrrrXrXrY_model_proto_reprvs&&rr c Cs|jr dt|jd}nd}|jrdt|jd}nd}|jr+dt|jd}nd}|jr:dt|jd }nd}|jrId t|jd }nd}d |jd |||||d S)Nz, initializer=r, node=< nodes>z, value_info= , input=< inputs> , output=< outputs>z GraphProto('rr) initializerrnode value_infoinputoutputrb)rrrrrrrXrXrY_graph_proto_reprs $rrc Cs|jr d|jd}nd}|jrd|jd}nd}|jr'dt|jd}nd}|jr3d|j}nd}|jrAdt|j}nd}|jrPd t|jd }nd}|jr_d t|jd }nd}d |j d|||||||d S)Nrrrz , overload='rrz , attribute=rrrrrzFunctionProto('r) roverloadrr attributerrrrrb)rrrrrrrrrXrXrY_function_proto_reprs,,rprotosSequence[OperatorSetIdProto]cCsdd|D}t|S)NcSsi|]}|j|jqSrX)rversion).0rrXrXrY sz-_operator_set_protos_repr..)repr)r opset_importsrXrXrYrsr)rIrJrKrL)rVrLrIrJrKr[)rIr`rKra)N)rjrkrIrlrKrm)NT)rIrJrwrkrxryrKr#)rIrJrwrkrKr2)rrrwrrKr#)rrLrwrrKr2)rrrIrJrwrkrryrryrrarrrryrKr[)rr2rIrJrwrkrKr[)rr#rKrd)rr rKrd)rrrKrd)rrrKrd)a __future__r__all__rerSrArBonnxrConnx.onnx_cpp2py_exportronnx.external_data_helperr9r@r8 onnx.onnx_pbrrrrr r!rrrrrr r r r r rr#r$r%r'r)rr,r-r.r/r0r1r2r3r4r5r6r7onnx.onnx_operators_pbr&r(onnx.onnx_data_pbr"r*r+ onnx.versionrrrrrrrrrrrr TYPE_CHECKINGcollections.abcrDr __version__rdrrrrZr_rirvr;r=r:r<r>r?loadload_from_stringsaverrrr__repr__rXrXrXrYsh F  %8      "   8