o e@sPddlZddlmZddlmZmZmZddlTddlTddl Tddl m Z ddl Z ddlZ ddlZ ddlZ ddlZ ddlZddlmZmZmZmZmZmZdeeeefd efd d Zd edeeeefd dfd dZdeeeefd eefddZdeeej j!j"fd efddZ#edej j!j"dZ$dede$d e$fddZ%d/deeeefdeede&d e'fddZ(d0deeeefdeed e)fddZ*d0dedeed e'fd d!Z+d0dedeed e)fd"d#Z,d1dee'efdeeeefdeed&e&d'e&d(eed)e-d*e&d dfd+d,Z.de)deeeefd dfd-d.Z/e(Z0e+Z1e.Z2dS)2N)ONNX_ML)load_external_data_for_modelwrite_external_data_tensorsconvert_model_to_external_data)*)version)UnionIOOptionalcastTypeVarAnyfreturncCsrt|drtttt|jrttt|}|Sttt|d }|}Wd|S1s2wY|S)Nreadrb)hasattrcallabler r bytesropenstr)rsreadabler=D:\Projects\ConvertPro\env\Lib\site-packages\onnx/__init__.py _load_bytess  rcontentcCsvt|drtttt|jrttt||dSttt|d}||WddS1s4wYdS)Nwritewb)rrr r rrrr)rrwritablerrr _save_bytes$s  "r!cCs2t|tr tj|St|drtj|jSdS)Nname) isinstancerospathabspathrr")rrrr_get_file_path-s   r'protocCs@t|tr|St|drt|jr|}|Stdt|)z Serialize a in-memory proto to bytes Arguments: proto: a in-memory proto, such as a ModelProto, TensorProto, etc Returns: Serialized proto in bytes SerializeToStringzKNo SerializeToString method is detected. neither proto is a str. type is {})r#rrrr) TypeErrorformattype)r(resultrrr _serialize5s  r._Proto)boundrcCst|tstdt|t|drt|js!tdt|tt t ||}|durB|t |krBt j jd|t ||S)z Parse bytes into a in-memory proto Arguments: s: bytes containing serialized proto proto: a in-memory proto object Returns: The proto instance filled in by s z)Parameter s must be bytes, but got type: ParseFromStringz2No ParseFromString method is detected. type is {}Nz6Protobuf decoding consumed too few bytes: {} out of {})r#r ValueErrorr,rrr1r+r r intlengoogleprotobufmessage DecodeError)rr(decodedrrr _deserializeLs  r:Tr+load_external_datacCs>t|}t||d}|rt|}|rtj|}t|||S)a Loads a serialized ModelProto into memory load_external_data is true if the external data under the same directory of the model and load the external data If not, users need to call load_external_data_for_model with directory to load Arguments: f: can be a file-like object (has "read" function) or a string containing a file name format: for future use Returns: Loaded in-memory ModelProto r+)rload_model_from_stringr'r$r%dirnamer)rr+r;rmodelmodel_filepathbase_dirrrr load_modelfs   rBcCst|}t||dS)z Loads a serialized TensorProto into memory Arguments: f: can be a file-like object (has "read" function) or a string containing a file name format: for future use Returns: Loaded in-memory TensorProto r<)rload_tensor_from_string)rr+rrrr load_tensors rDcC t|tS)z Loads a binary string (bytes) that contains serialized ModelProto Arguments: s: a string, which contains serialized ModelProto format: for future use Returns: Loaded in-memory ModelProto )r: ModelProtorr+rrrr= r=cCrE)z Loads a binary string (bytes) that contains serialized TensorProto Arguments: s: a string, which contains serialized TensorProto format: for future use Returns: Loaded in-memory TensorProto )r: TensorProtorGrrrrCrHrCFsave_as_external_dataall_tensors_to_one_filelocationsize_thresholdconvert_attributec Csbt|tr t|t}|rt|||||t|}|r&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. Arguments: proto: should be a in-memory ModelProto f: can be a file-like object (has "write" function) or a string containing a file name format for future use all_tensors_to_one_file: 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: specify the external file that all tensors to save to. If not specified, will use the model name. size_threshold: 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: If true, convert all tensors to external data If false, convert only non-attribute tensors to external data N) r#rr:rFrr'r$r%r>rr.r!) r(rr+rKrLrMrNrOr@Zbasepathrrrr save_models    rPcCst|}t||dS)z Saves the TensorProto to the specified path. Arguments: proto: should be a in-memory TensorProto f: can be a file-like object (has "write" function) or a string containing a file name format: for future use N)r.r!)r(rrrrr save_tensors rQ)NT)N)NFTNrJF)3r$Zonnx_cpp2py_exportrZonnx.external_data_helperrrrZonnx_pbZonnx_operators_pbZ onnx_data_pbr __version__Z onnx.checkerZonnxZ onnx.defsZ onnx.helperZ onnx.utilsZ onnx.composeZgoogle.protobuf.messager5typingr r r r r rrrrr!r'r6r7Messager.r/r:boolrFrBrIrDr=rCr3rPrQloadZload_from_stringsaverrrrs:   " " ,(L"