o soi@s,UddlmZddlZddlZddlZddlZddlZddlmZm Z m Z ddl Z ddl ZddlmZddlZddlZddlmZmZddlmZmZmZddlmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$eryddlm%Z%m&Z&m'Z'ddl(m)Z)e*e+e,e,e,fe*e+e,e,e,e,fBZ-e.e-Z/e.e*e+e+fZ0gd Z1d e2d <e3e*e+e,fe,fZ4dddZ5e5e1Z6 ddddZ7    ddd&d'Z8dd*d+Z9    ddd6d7Z:dd8d9Z;     dddAdBZddNdOZ?ddQdRZ@ddVdWZAddXdYZB dddadbZCddfdgZDddldmZEddrdsZFddxdyZGdd|d}ZH  ddddZI ddddZJdddZKdddZLdddZM ddddZN ddddZO ddddZP ddddZQdddZRdddZSdddZT d d ddZUd ddZV dd ddZW dd ddZXdddZYdddZZdddZ[dddZ\ dddd„Z]e^dád dddńZ_dddȄZ`ddd҄ZadddքZbddd؄ZcdddڄZdeedddd܄ZfeeddddZgdddZhddejijDZkde2d<dddZldS() annotationsN) TYPE_CHECKINGAnyTypeVar)_mappingdefs)MapProto OptionalProto SequenceProto) AttributeProto FunctionProto GraphProto ModelProto NodeProtoOperatorSetIdProto TensorProtoTensorShapeProtoTrainingInfoProto TypeProtoValueInfoProto)CallableKeysViewSequence)RepeatedCompositeFieldContainer))z1.0r)z1.1rr)z1.1.2rr)z1.2rr)z1.3rr)z1.4.1 r)z1.5.0r r)z1.6.0r )z1.7.0r r$r)z1.8.0r r$r)z1.8.1rr&r$r)z1.9.0rr$r)z1.10.0rr$r)z1.10.1rr(r$r)z1.10.2rr(r$r)z1.11.0rrr)z1.12.0rrr)z1.13.0rrr)z1.13.1rr+rr)z1.14.0r!rr)z1.14.1r!r,rr)z1.15.0r!r r)z1.16.0r"rr)z1.16.1r"r.rr)z1.16.2r"r.rr)z1.17.0r"rr)z1.18.0r#rr)z1.19.0r%rr)z1.19.1r%r1rr)z1.20.0r&rr)z1.20.1r&r2rrVersionTableType VERSION_TABLEtablereturnVersionMapTypecs(id fd d }|D]}||q S) zOCreate a map from (opset-domain, opset-version) to ir-version from above table.release_versionstr ir_versionintargsrr6NonecsJ~tgd|ddD]}|vr"||<|ddkr"|d|df<q dS)N)ai.onnxz ai.onnx.mlai.onnx.trainingF)strictrr?zai.onnx.preview.trainingr)zip)r8r:r<pairresult?/home/ubuntu/.local/lib/python3.10/site-packages/onnx/helper.pyprocessXs   z._create_op_set_id_version_map..processN)r8r9r:r;r<rr6r=rE)r5rGrowrErCrF_create_op_set_id_version_mapTs  rIF opsetidlistSequence[OperatorSetIdProto]ignore_unknownboolr;cs2dd fdd |rtfd d |DSS) a=Given list of opset ids, determine minimum IR version required. Args: opsetidlist: A sequence of OperatorSetIdProto. ignore_unknown: If True, ignore unknown domain and return default minimum version for that domain. Returns: The minimum IR version required (integer) rdomain str | Noneversionr;r6cs,|pd|f}|tvrt|SrStd)Nr>zUnsupported opset-version.)OP_SET_ID_VERSION_MAP ValueError)rNrPkey)default_min_versionrLrErFfind_minys z)find_min_ir_version_for..find_minc3s|] }|j|jVqdSN)rNrP.0x)rUrErF sz*find_min_ir_version_for..N)rNrOrPr;r6r;)max)rJrLrE)rTrUrLrFfind_min_ir_version_forjs  r\op_typer9inputs Sequence[str]outputsnamerO doc_stringrNoverloadkwargsrrc Kszt}||_|j||j||r||_|r||_|dur#||_|dur*||_|r;|j ddt | D|S)aConstruct a NodeProto. Args: op_type (string): The name of the operator to construct inputs (list of string): list of input names outputs (list of string): list of output names name (string, default None): optional unique identifier for NodeProto doc_string (string, default None): optional documentation string for NodeProto domain (string, default None): optional domain for NodeProto. If it's None, we will just use default domain (which is empty) overload (string, default None): optional field, used to resolve calls to model-local functions **kwargs (dict): the attributes of the node. The acceptable values are documented in :func:`make_attribute`. Returns: NodeProto Ncss&|]\}}|durt||VqdSrV)make_attribute)rXrSvaluerErErFrZszmake_node..) rr]inputextendoutputrarbrNrc attributesorteditems) r]r^r`rarbrNrcrdnoderErErF make_nodes"    rnrPrcCt}||_||_|SzConstruct an OperatorSetIdProto. Args: domain (string): The domain of the operator set id version (integer): Version of operator set id Returns: OperatorSetIdProto rrNrP)rNrP operatorsetidrErErFmake_operatorsetids rsnodesSequence[NodeProto]Sequence[ValueInfoProto] initializerSequence[TensorProto] | None value_infoSequence[ValueInfoProto] | Nonesparse_initializer'Sequence[onnx.SparseTensorProto] | Noner c Cs|durg}|dur g}|durg}t}|j|||_|j||j||j||j||j||rA||_ |S)aConstruct a GraphProto Args: nodes: list of NodeProto name (string): graph name inputs: list of ValueInfoProto outputs: list of ValueInfoProto initializer: list of TensorProto doc_string (string): graph documentation value_info: list of ValueInfoProto sparse_initializer: list of onnx.SparseTensorProto Returns: GraphProto N) r rmrhrargrirwr{ryrb) rtrar^r`rwrbryr{graphrErErF make_graphs"      r~cCrorprq)rNrPopsetidrErErF make_opsetids rfname opset_imports attributesSequence[str] | Noneattribute_protosSequence[AttributeProto] | Noner c Cs|durg}|dur g}| durg} t} || _|| _| j|| j|| j|| j|| j|| j ||rD|| _ | durK| | _ | j | | SrV) r rNrargrhrirm opset_importrjattribute_protorbrcry) rNrr^r`rtrrrrbrcryfrErErF make_functions*        rr}rcKst}tj|_|j||dd}|dur|j|n |j }t |_ |dd}|dur8|j ||D] \}}t|||q<|S)zConstruct a ModelProto Args: graph (GraphProto): *make_graph* returns **kwargs: any attribute to add to the returned instance Returns: ModelProto rN functions)ronnx IR_VERSIONr:r}CopyFrompoprrhaddronnx_opset_versionrPrrlsetattr)r}rdmodelrimprkvrErErF make_model's       rcKs8d}||vrd}||g}t|||<t|fi|S)Nr:r)getr\r)r}rdir_version_fieldopset_imports_fieldimportsrErErFmake_model_gen_versionLs   rprotoRModelProto | GraphProto | FunctionProto | NodeProto | TensorProto | ValueInfoProto dict_valuedict[str, str]r=cCs8|jdd=|D]\}}|j}||_||_q dSrV)metadata_propsrlrrSrf)rrrrentryrErErFset_metadata_propsUs  rrcCst||dSrV)r)rrrErErFset_model_propsgsrarray np.ndarraynpt.NDArray[np.uint8]cCsx|tj}|j}|ddk}|r|j|dgdd|dM}|ddddK<|ddd|dddBS) z[Convert a numpy array to flatten, packed int4/uint4. Elements must be in the correct range.r$rFrefcheckr(Nr rravelviewnpuint8copysizeresize)r array_flatr odd_sizedrErErF _pack_4bitx2ks rcCs|tj}|j}|d}|r|j|d|gdd|dM}|ddddK<|ddddK<|ddddK<|d dd|dddB|dddB|dddBS) z[Convert a numpy array to flatten, packed int2/uint2. Elements must be in the correct range.r FrrrNr$rrr)rrrpad_lenrErErF _pack_2bitx4xs8r data_typedims Sequence[int]vals*Sequence[int | float] | bytes | np.ndarrayrawrc Cst}||_||_|j||tjkr|rtdt|}|r|tjtj tj hvr-d}n|tj tj hvr8d}n|j }|t|9}t|}t|tjrs|tj tjtj hvr^tj|}n|tj tj hvrltj|}tj|}nt|tr{|}n tdt|dt||krtd|dt|d||_|S|rJd |tjkrt|}t|d krt t!|}n6|tj"tj#tj$tj%hvrtj&t'||}n|tj(krtjj)t'|d d d }n tj'||d}|tj*kr|+tj,}n]|tj-kr |+tj.}nP|tj/tj0hvr|+tj1}n@|tj"tj#tj$tj%tj(hvr2|+tj2}n*|tjtj tj hvrBt|}n|tj tj hvrPt|}n |tj3kr\|4tj2}t5|} t6|| ||S)auMake a TensorProto with specified arguments. If raw is False, this function will choose the corresponding proto field to store the values based on data_type. If raw is True, use "raw_data" proto field to store the values, and values should be of type bytes in this case. Args: name: tensor name data_type: a value such as onnx.TensorProto.FLOAT dims: shape vals: values raw: if True, vals contains the serialized content of the tensor, otherwise, vals should be a list of values of the type defined by ``data_type``. Returns: TensorProto z*Can not use raw_data to store string type.g?g?z1Raw data must be bytes or numpy.ndarray, but got .z5Raw data size does not match tensor's size. Expected z bytes, but got z bytes.z'Bug: raw should be False at this point.rTup)saturate round_mode)dtype)7rrrarrhSTRING TypeErrortensor_dtype_to_np_dtypeUINT4INT4 FLOAT4E2M1UINT2INT2itemsizemathprodceil isinstancerndarrayr numpy_helperrrtobytes_little_endianbytestypelenrRraw_datarflatten vectorize _to_bytes FLOAT8E4M3FNFLOAT8E4M3FNUZ FLOAT8E5M2FLOAT8E5M2FNUZ saturate_castasarray FLOAT8E8M0 to_float8e8m0 COMPLEX128rfloat64 COMPLEX64float32BFLOAT16FLOAT16uint16rBOOLastypetensor_dtype_to_fieldgetattr) rarrrrtensornp_dtypeexpected_size_bytesrfieldrErErF make_tensors                 rvaluesindicesonnx.SparseTensorProtocCs0t}|j||j||j||S)zConstruct a SparseTensorProto Args: values (TensorProto): the values indices (TensorProto): the indices dims: the shape Returns: SparseTensorProto )rSparseTensorProtorrrrrh)rrrsparserErErFmake_sparse_tensors    r elem_typeSequenceProto.DataType Sequence[Any]r cCst}||_||_|tjkr|Sd}|tjkr|j}n(|tjkr$|j}n|tjkr-|j }n|tj kr6|j }n |t j kr?|j}ntd|||S)z/Make a Sequence with specified value arguments.Nz8The element type in the input sequence is not supported.)r rar UNDEFINEDTENSOR tensor_values SPARSE_TENSORsparse_tensor_valuesSEQUENCEsequence_valuesMAP map_valuesr OPTIONALoptional_valuesrrh)rarrsequencerjrErErF make_sequences&       rkey_typekeys list[Any]rcCsrt}tjtjtjtjtjtjtjtj g}||_ ||_ |tj kr'|j |n ||vr1|j||j||S)zMake a Map with specified key-value pair arguments. Criteria for conversion: - Keys and Values must have the same number of elements - Every key in keys must be of the same type - Every value in values must be of the same type )rrINT8INT16INT32INT64UINT8UINT16UINT32UINT64rarr string_keysrhrrr)rarrr map_protovalid_key_int_typesrErErFmake_map/s$     rOptionalProto.DataTyperf&google.protobuf.message.Message | Noner cCst}||_||_|tjkr|Sd}|tjkr|j}n(|tjkr$|j}n|tjkr-|j }n|tj kr6|j }n |tj kr?|j }ntd|dusIJ|||S)z0Make an Optional with specified value arguments.Nz8The element type in the input optional is not supported.)r rarrr tensor_valuersparse_tensor_valuersequence_valuer map_valueroptional_valuerr)rarrfoptionalrjrErErF make_optionalNs(        r str | bytesrcCst|tr |dS|S)z2Coerce a string (or bytes) value into UTF-8 bytes.utf-8)rr9encode)rfrErErFrms  rrS attr_type int | Noner c s(t}||_|r ||_t|tjrt||_tj|_ nTt|tj r-t ||_ tj |_ nCt|ttfr?t||_tj|_ n1t|trP|j|tj|_ n t|tjrb|j|tj|_ nt|trr|j|tj|_ nt|tr|j|tj |_ nt|t!j"j#rht$|}t%|dkr|durt&d|d|durdd|D}tjtj'ftj tj(fttftj)fttj*ftjtj+fttj,fttj-ffD]\}t.fdd|Dr|}nq|durt&d |tj'kr|j/0|tj'|_ nv|tj(kr |j10|tj(|_ ne|tj)kr!|j20d d|Dtj)|_ nO|tj*kr2|j30|tj*|_ n>|tj+krC|j40|tj+|_ n-|tj,krT|j50|tj,|_ n|tj-kre|j60|tj-|_ n t7t8d |d |dur|j |krt8d t9|j d|j dt9|d|d |S)z0Makes an AttributeProto based on the value type.rNzCould not infer attribute `z` type from empty iteratorcSsh|]}t|qSrE)rrXrrErErF sz!make_attribute..c3s|]}t|VqdSrV) issubclassrXtexp_trErFrZsz!make_attribute..zRCould not infer the attribute type from the elements of the passed Iterable value.css|]}t|VqdSrV)rr$rErErFrZs'z%' is not an accepted attribute value.zInferred attribute type 'z'(z") mismatched with specified type ')):r rarbrnumbersIntegralr;iINTrRealfloatrFLOATr9rrsrrr(rrrr sparse_tensorrr gGRAPHrtp TYPE_PROTO collectionsabcIterablelistrrRINTSFLOATSSTRINGSTENSORSSPARSE_TENSORSGRAPHS TYPE_PROTOSallintsrhfloatsstringstensorssparse_tensorsgraphs type_protosAssertionErrorr_attr_type_to_str)rSrfrbr"attrtypesexp_enumrEr)rFrets                                               (reAttributeProto.AttributeTypecCs t}||_||_|r||_|S)zeMake an AttributeProto holding a reference to the parent function's attribute of given name and type.)r rarrb)rar"rbrOrErErFmake_attribute_refs rSrOcCsJ|jr td||jtjkr|jS|jtjkr|jS|jtjkr%|j S|jtj kr.|j S|jtj kr7|j S|jtjkr@|jS|jtjkrI|jS|jtjkrTt|jS|jtjkr_t|jS|jtjkrjt|jS|jtjkrut|jS|jtjkrt|jS|jtjkrt|jS|jtjkrt|j S|jtj!krdStd|)Nz)Cannot get value of reference attribute: zUnsupported ONNX attribute: )" ref_attr_namerRrr r3rr0r/rr4rr(rr5r7r6r9r8r?r=rGr>rFr@rHrArIrBrJrCrKrDrLr)rOrErErFget_attribute_valuesB                      rUrm attr_namecsTfdd|jD}t|dkrtdt|dkr$tdt|dS)Ncsg|] }|jkr|qSrErarWrVrErF sz'get_node_attr_value..rz'Node has multiple attributes with name z Node has no attribute with name r)rjrrRrU)rmrVmatchingrErXrFget_node_attr_values    r[rcCst}||_|SrV)rra)ravalue_info_protorErErFmake_empty_tensor_value_info sr]shape!Sequence[str | int | None] | Noneshape_denotationlist[str] | Nonerc Ct}|j}||_|j}|durX|jg|r$t|t|kr$tdt|D]/\}}|j }|dur6nt |t r?||_ nt |t rH||_ntd|d|rW|||_q(|S)z:Makes a Tensor TypeProto based on the data type and shape.N>Invalid shape_denotation. Must be of the same length as shape.Invalid item in shape: z. Needs to be of int or str.)r tensor_typerr^dimrhrrR enumeraterrr; dim_valuer9 dim_param denotation) rr^r` type_prototensor_type_prototensor_shape_protor/drfrErErFmake_tensor_type_proto2      rocCs2t}||_|r ||_t|||}|j||S)z8Makes a ValueInfoProto based on the data type and shape.)rrarbrorr)rarr^rbr`r\rlrErErFmake_tensor_value_info>s  rrc Crb)z@Makes a SparseTensor TypeProto based on the data type and shape.Nrcrdz. Needs to be of int or text.)rsparse_tensor_typerr^rfrhrrRrgrrr;rhr9rirj) rr^r`rksparse_tensor_type_protosparse_tensor_shape_protor/rnrfrErErFmake_sparse_tensor_type_protoPrprvcCs6t}||_|r ||_t|||}|jj|j|S)zEMakes a SparseTensor ValueInfoProto based on the data type and shape.)rrarbrvrrsr)rarr^rbr`r\rtrErErFmake_sparse_tensor_value_info}srwinner_type_protocCt}|jj||S)zMakes a sequence TypeProto.)r sequence_typerrrxrkrErErFmake_sequence_type_protor|cCry)zMakes an optional TypeProto.)r optional_typerrr{rErErFmake_optional_type_protor}r value_typecCs t}||j_|jj||S)zMakes a map TypeProto.)rmap_typerrr)rrrkrErErFmake_map_type_protosrrkcCs&t}||_|r ||_|j||S)z1Makes a ValueInfoProto with the given type_proto.)rrarbrr)rarkrbr\rErErFmake_value_infos  rr4cCsbt|tr|}nt|tr|jddd}nt|}t|dkr!|S|dddt|ddS)Nr ignore)errors@z ...<+len=>)rr9rdecoder)r4 sanitizedrErErF _sanitize_strs    relem_shape_denotationcCs>t}||_|r ||_t|||}t|}|jj|j|S)zJMakes a Sequence[Tensors] ValueInfoProto based on the data type and shape.)rrarbror|rrzr)rarr^rbrr\rlsequence_type_protorErErFmake_tensor_sequence_value_infos r subgraphs"str | tuple[str, list[GraphProto]]c Csg}||j|dd%dd}d&d d }td }d'dd}g}|dr1|||jn|dr@|||jn|drP|tt|jn|drwt |j j dkrc|dnt |j j }|dt|j |dn|dr|d|jjd||jn|dr|d|jdn|jr||||jn|jr||||jn|jr|tttt|jny|jr|dnp|jr|dt|jD]\} } | t |jdkrd nd!} |d| d| q|d"n?|jr=|dt|jD]\} } | t |jdkr!d nd!} |d| jd| q|d"||jn|d#|rLd$||fSd$|S)(N=rr2r6r9cSs|dS)Nz.15grE)rrErErF str_floatsz&printable_attribute..str_floatr/r;cSst|SrV)r9)r/rErErFstr_intsz$printable_attribute..str_int_Tstr_elemCallable[[_T], str]xs Sequence[_T]cSsddt||dS)N[, ])joinmap)rrrErErFstr_listsz%printable_attribute..str_listr4r(rzz, ...]rr,rqrz  )rr2r6r9)r/r;r6r9)rrrrr6r9)appendrarHasFieldrr/reprrr4rr(rrrrr6r8rGrFrHr9r=rrIrLrgrKrhr) rOrcontentrrrrrKrr/r8commar6rErErFprintable_attributesb                    rrfTensorShapeProto.DimensioncCs$|d}|dur dStt||S)Nrf?) WhichOneofr9r)rfwhichrErErF printable_dim+s rr(cCs|ddkr5tj|jj}|jdr3t|jjj r/|t dd t t |jjj 7}|S|d7}|S|ddur>dSd|dS) Nrfrer^rrY, scalarrqz Unknown type )rrDataTypeNamererrrr^rfr9rrrr(r4rErErFprintable_type2s "rrcCs,d|j}|jr|dt|jd}|S)N%rr)rarr)rr4rErErFprintable_value_info@s rcCshd|jd}|tj|j7}|jdur.t|jr*|tddt t|j7}n|d7}|d7}|S)NrrrrYrr) rarrrrrrr9rrrrErErFprintable_tensor_protoGs   rprefixc CsDg}t|jr|ddd|jD|dg}g}|jD]=}|rHt||}t|dts9tdtd| |d||dq!t|}t|t sYtd t d||q!dt |} dd d|j D} |jr||j d | d | d n ||j d| d |r|d||fS|d|S)NrcSg|]}d|qSrrErXrarErErFrYXz"printable_node..rrz1printed_attr_subgraphs[1] must be an instance of rrzprinted must be an instance of cSrrrErrErErFrYlrrz](r,(r)rrirrrjrrr=rrhr9rkrgr]) rmrrrrK printed_attrsrOprinted_attr_subgraphsprintedprinted_attributesprinted_inputsrErErFprintable_nodeSs6        rzEDeprecated since 1.19. Consider using onnx.printer.to_text() instead.csng}|d}d|jg}dd|jD}t|jr|dg}g}|jD]}|j|vr3|t|q$|t|q$|rW||d|g}|D] } ||d| qK|d|r|d||d|g}|D] } ||d| qq|dt|t|krd d|jDfd d |jD} |d ||d|g}| D] } ||d| q|d|d ||d|g} |jD]&} t| |dd} t | dt st dt d|| d| | dqdg}t|j r|ddd |j D||d|||d| D] }|dt|q%d|S)aDisplay a GraphProto as a string. .. deprecated:: 1.19 Consider using :func:`onnx.printer.to_text` instead. Args: graph (GraphProto): the graph to display prefix (string): prefix of every line Returns: string z r}cSh|]}|jqSrErWr'rErErFr%z"printable_graph..rrr,z,optional inputs with matching initializers (cSrrErWrXr/rErErFr%rcsg|] }|jvrt|qSrE)rarr graph_inputsrErFrYs  z#printable_graph..zinitializers ({T)rrz-contents_subgraphs[1] must be an instance of rrr6rcSsg|]}d|jqSrrW)rXoutrErErFrYs} )rarwrrgrrrrmrrr=rrhriprintable_graph)r}rrindentheader initializersin_strsin_with_init_strsinpline init_strsrKrmcontents_subgraphstailr6rErrFrvsj               rgoogle.protobuf.message.MessagecCst|tjjjstdtjjjd|jjD]6}|jdkr%| |jq|j |j krM|j |j kr?t||jD]}t|q7q||jrMtt||jqdS)z:Empties `doc_string` field on any nested protobuf messageszproto must be an instance of rrbN)rgoogleprotobufmessageMessager DESCRIPTORfieldsra ClearFieldr TYPE_MESSAGElabelLABEL_REPEATEDrstrip_doc_stringr)r descriptorrYrErErFrs       r algorithmalgorithm_bindingsAssignmentBindingTypeinitializationGraphProto | Noneinitialization_bindingsAssignmentBindingType | NonercCsrt}|j||D]\}}|j}||_||_q |r#|j||r7|D]\}}|j}||_||_q'|SrV) rrrupdate_bindingrrSrfrinitialization_binding)rrrr training_inforrbindingrErErFmake_training_infos      r tensor_dtypenp.dtypecC tj|jS)zConvert a TensorProto's data_type to corresponding numpy dtype. It can be used while making tensor. Args: tensor_dtype: TensorProto's data_type Returns: numpy's data_type )rTENSOR_TYPE_MAPrrrErErFr rcCr)zConvert a TensorProto's data_type to corresponding data_type for storage. Args: tensor_dtype: TensorProto's data_type Returns: data_type for storage )rr storage_dtyperrErErF$tensor_dtype_to_storage_tensor_dtyperrcCr)zGet the name of given TensorProto's data_type. Args: tensor_dtype: TensorProto's data_type Returns: the name of data_type )rrrarrErErFtensor_dtype_to_string rrcCsZttjdttjdttjdttjdttjdttjdttjdi}|t j |j S)zConvert a TensorProto's data_type to corresponding field name for storage. It can be used while making tensors. Args: tensor_dtype: TensorProto's data_type Returns: field name float_data int32_data int64_data double_data uint64_data string_data) r;rr3r r DOUBLErrrrrr)rstorage_tensor_type_to_fieldrErErFrs       rrTensorProto.DataTypecCsPddtjD}||vrtd||St|tjr tj St d|d)zConvert a numpy's dtype to corresponding tensor type. It can be used while converting numpy arrays to tensors. Args: np_dtype: numpy's data_type Returns: TensorsProto's data_type cSsi|]\}}|j|qSrE)r)rXrrrErErF 9s z,np_dtype_to_tensor_dtype..rzUnable to convert type z into TensorProto element type.) rrrltypingcastr issubdtypestr_rrrR)r_np_dtype_to_tensor_dtyperErErFnp_dtype_to_tensor_dtype/s  r KeysView[int]cCs tjS)zcGet all tensor types from TensorProto. Returns: all tensor types from TensorProto )rrrrErErErFget_all_tensor_dtypesFs rcCsi|]\}}||qSrErE)rXrrrErErFrOsrzdict[int, str]_ATTRIBUTE_TYPE_TO_STRcCs$|tjvr t|StjdS)zConvert AttributeProto type to string. Args: attr_type: AttributeProto type. Returns: String representing the supplied attr_type. r)r AttributeTyperrr)r"rErErFrNUs rN)r5r3r6r7)F)rJrKrLrMr6r;)NNNN)r]r9r^r_r`r_rarOrbrOrNrOrcrOrdrr6r)rNr9rPr;r6r)rtrurar9r^rvr`rvrwrxrbrOryrzr{r|r6r )NNNNN)rNr9rr9r^r_r`r_rtrurrKrrrrrbrOrcrOryrzr6r )r}r rdrr6r)rrrrr6r=)rrrrr6r=)rrr6r) rar9rr;rrrrrrMr6r)rrrrrrr6r)rar9rrrrr6r ) rar9rr;rr rr r6r)rar9rrrfrr6r )rfrr6r)NN) rSr9rfrrbrOr"r#r6r rV)rar9r"rRrbrOr6r )rOr r6r)rmrrVr9r6r)rar9r6r)rr;r^r_r`rar6r)rqN) rar9rr;r^r_rbr9r`rar6r)rxrr6r)rr;rrr6r)rq)rar9rkrrbr9r6r)r4rr6r9) rar9rr;r^r_rbr9rrar6r)rOr rrMr6r)rfrr6r9)r(rr6r9)rrr6r9)r(rr6r9)rqF)rmrrr9rrMr6r)r}r rr9r6r9)rrr6r=) rr rrrrrrr6r)rr;r6r)rr;r6r;)rr;r6r9)rrr6r)r6r)r"r;r6r9)m __future__rcollections.abcr: functoolsrr-rrrrgoogle.protobuf.messagernumpyr numpy.typingnpttyping_extensionsrrronnx.onnx_data_pbrr r onnx.onnx_pbr r r rrrrrrrrrrr#google.protobuf.internal.containersrtupler9r;VersionRowTyper=r3rr4__annotations__dictr7rIrQr\rnrsr~rrrrrrrrrrrrrrrerSrUr[r]rorrrvrwr|rrrrrrrrrrr deprecatedrrrrrr lru_cacherrrrrlrrNrErErErFs 4 "#    1 + $ %    u      _ $  11    L    # P