o q::jK@sUddlmZddlZddlZddlZddlZddlZddlmZm Z m Z ddl Z ddl ZddlZddlZddlmZmZddlmZmZmZddlmZmZmZmZmZmZmZmZm Z m!Z!m"Z"ersddlm#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)fZ.gd Z/d e0d <e1e(e)e*fe*fZ2dddZ3e3e/Z4 ddddZ5    ddd&d'Z6dd*d+Z7    ddd6d7Z8dd8d9Z9     dddAdBZ:ddEdFZ;ddGdHZ dddZd[Z?dd_d`Z@ddedfZAddkdlZBddqdrZCddudvZD  ddd{d|ZE ddd}dddZFdddZGdddZHdddZI ddddZJ ddddZK ddddZL ddddZMdddZNdddZOdddZP ddddZQdddZR ddddZS ddddZTdddZUd ddZVd ddZWd ddZX d d ddZYeZdddddZ[dddÄZ\ddd̈́Z]dddфZ^dddӄZ_dddՄZ`eaddddׄZbeaddddۄZcdddބZdddejefDZgde0d<dddZhdS() 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&r2rr)z1.21.0r&rrVersionTableType VERSION_TABLEtablereturnVersionMapTypecCsi}|D]:}|d}tgd|ddddD]&}||vr=|||<|ddkr/||d |df<|dd kr=||d |df<qq|S) zOCreate a map from (opset-domain, opset-version) to ir-version from above table.r)ai.onnxz ai.onnx.mlai.onnx.trainingr$NF)strictrr9zai.onnx.previewr:zai.onnx.preview.training)zip)r6resultrow ir_versionpairrAR/home/nk/hobo-godmode/plappi-mvp/.venv/lib/python3.10/site-packages/onnx/helper.py_create_op_set_id_version_mapTs"     rCF opsetidlistSequence[OperatorSetIdProto]ignore_unknownboolintcs2dd 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 | NoneversionrHr7cs,|pd|f}|tvrt|SrStd)Nr9zUnsupported opset-version.)OP_SET_ID_VERSION_MAP ValueError)rIrKkey)default_min_versionrFrArBfind_minys z)find_min_ir_version_for..find_minc3s|] }|j|jVqdSN)rIrK.0x)rPrArB sz*find_min_ir_version_for..N)rIrJrKrHr7rH)max)rDrFrA)rOrPrFrBfind_min_ir_version_forjs  rWop_typestrinputs Sequence[str]outputsnamerJ doc_stringrIoverloadkwargsrrc 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||VqdSrQ)make_attribute)rSrNvaluerArArBrUszmake_node..) rrXinputextendoutputr]r^rIr_ attributesorteditems) rXrZr\r]r^rIr_r`noderArArB make_nodes"    rjrKrcCt}||_||_|SzConstruct an OperatorSetIdProto. Args: domain (string): The domain of the operator set id version (integer): Version of operator set id Returns: OperatorSetIdProto rrIrK)rIrK operatorsetidrArArBmake_operatorsetids ronodesSequence[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 rirdr]rcrersrwrur^) rpr]rZr\rsr^rurwgraphrArArB make_graphs"      rzcCrkrlrm)rIrKopsetidrArArB make_opsetids r|fname opset_imports attributesSequence[str] | Noneattribute_protosSequence[AttributeProto] | Noner c Cs|durg}|dur g}| durg} t} || _|| _| j|| j|| j|| j|| j|| j ||rD|| _ | durK| | _ | j | | SrQ) r rIr]rcrdreri opset_importrfattribute_protor^r_ru) rIr}rZr\rpr~rrr^r_rufrArArB make_functions*        rryrcKst}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 r~N functions)ronnx IR_VERSIONr?ryCopyFrompoprrdaddronnx_opset_versionrKrrhsetattr)ryr`modelr~imprkvrArArB make_model's       rcKs8d}||vrd}||g}t|||<t|fi|S)Nr?r~)getrWr)ryr`ir_version_fieldopset_imports_fieldimportsrArArBmake_model_gen_versionLs   rprotoRModelProto | GraphProto | FunctionProto | NodeProto | TensorProto | ValueInfoProto dict_valuedict[str, str]NonecCs8|jdd=|D]\}}|j}||_||_q dSrQ)metadata_propsrhrrNrb)rrrrentryrArArBset_metadata_propsUs  rrcCst||dSrQ)r)rrrArArBset_model_propsgsr 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}t|} t|| kr tdt|d| d|tj*kr|+tj,}na|tj-kr&|+tj.}nT|tj/tj0hvr6|+tj1}nD|tj"tj#tj$tj%tj(hvrL|+tj2}n.|tjtj tj hvr^tj|}n|tj tj hvrntj|}n |tj3krz|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)dtypezNumber of values (z-) does not match tensor dimensions requiring z elements.)7rrr]rrdSTRING TypeErrortensor_dtype_to_np_dtypeUINT4INT4 FLOAT4E2M1UINT2INT2itemsizemathprodceil isinstancenpndarrayr numpy_helper _pack_4bitx2 _pack_2bitx4tobytes_little_endianbytestypelenrMraw_dataarrayflatten vectorize _to_bytes FLOAT8E4M3FNFLOAT8E4M3FNUZ FLOAT8E5M2FLOAT8E5M2FNUZ saturate_castasarray FLOAT8E8M0 to_float8e8m0 COMPLEX128viewfloat64 COMPLEX64float32BFLOAT16FLOAT16uint16uint8BOOLastypetensor_dtype_to_fieldgetattr) r]rrrrtensornp_dtypeexpected_size_bytesrexpected_elementsfieldrArArB make_tensorks                 rvaluesindicesonnx.SparseTensorProtocCs0t}|j||j||j||S)zConstruct a SparseTensorProto Args: values (TensorProto): the values indices (TensorProto): the indices dims: the shape Returns: SparseTensorProto )rSparseTensorProtorrrrrd)rrrsparserArArBmake_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 r]r UNDEFINEDTENSOR tensor_values SPARSE_TENSORsparse_tensor_valuesSEQUENCEsequence_valuesMAP map_valuesr OPTIONALoptional_valuesrrd)r]rrsequencerfrArArB 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 )rrINT8INT16INT32INT64UINT8UINT16UINT32UINT64r]rr string_keysrdrrr)r]rrr map_protovalid_key_int_typesrArArBmake_maps$     rOptionalProto.DataTyperb&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 r]rrr tensor_valuersparse_tensor_valuersequence_valuer map_valueroptional_valuerr)r]rrboptionalrfrArArB make_optional8s(        r str | bytesrcCst|tr |dS|S)z2Coerce a string (or bytes) value into UTF-8 bytes.utf-8)rrYencode)rbrArArBrWs  rrN attr_type int | Noner c s&t}||_|r ||_t|tjrt||_tj|_ nSt|tj r-t ||_ tj |_ nBt|ttfr?t||_tj|_ n0t|trP|j|tj|_ nt|tjrb|j|tj|_ n t|trr|j|tj|_ nt|tr|j|tj |_ nt|t!j"j#rgt$|}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'|_ nu|tj(kr |j10|tj(|_ nd|tj)kr!|j20d d|Dtj)|_ nN|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|qSrA)rrSrrArArB sz!make_attribute..c3s|]}t|VqdSrQ) issubclassrStexp_trArBrUsz!make_attribute..zRCould not infer the attribute type from the elements of the passed Iterable value.css|]}t|VqdSrQ)rrrArArBrUs'z%' is not an accepted attribute value.zInferred attribute type 'z'(z") mismatched with specified type ')):r r]r^rnumbersIntegralrHiINTrRealfloatrFLOATrYrrsrrrrrrr sparse_tensorrr gGRAPHrtp TYPE_PROTO collectionsabcIterablelistrrMINTSFLOATSSTRINGSTENSORSSPARSE_TENSORSGRAPHS TYPE_PROTOSallintsrdfloatsstringstensorssparse_tensorsgraphs type_protosAssertionErrorr_attr_type_to_str)rNrbr^rattrtypesexp_enumrArrBra^s                                               (ra) ref_attr_nameAttributeProto.AttributeTyperDcCs>|dur|}|s tdt}||_||_||_|r||_|S)aMake an AttributeProto holding a reference to the parent function's attribute. The returned attribute carries no value of its own; at instantiation time its value is supplied by the parent function's attribute named ``ref_attr_name``. When ``ref_attr_name`` is not provided, it defaults to ``name``. Reference attributes are only valid inside a function (sub-graph). Args: name: The name of this attribute as used inside the function body. attr_type: The type of the attribute. doc_string: Optional human-readable documentation for the attribute. ref_attr_name: The name of the parent function's attribute being referenced. Nzref_attr_name must be non-empty)rMr r]rrDr^)r]rr^rDrArArArBmake_attribute_refsrFrAcCsJ|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: )"rDrMrr r%rr"r!rr&rrrr'r)r(r+r*r1r/r9r0r8r2r:r3r;r4r<r5r=r6r>r)rArArArBget_attribute_valuesB                      rGri attr_namecsTfdd|jD}t|dkrtdt|dkr$tdt|dS)Ncsg|] }|jkr|qSrAr]rRrHrArB sz'get_node_attr_value..rz'Node has multiple attributes with name z Node has no attribute with name r)rfrrMrG)rirHmatchingrArJrBget_node_attr_values    rMrcCst}||_|SrQ)rr])r]value_info_protorArArBmake_empty_tensor_value_info srOshape!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_typerrPdimrdrrM enumeraterrrH dim_valuerY dim_param denotation) rrPrR type_prototensor_type_prototensor_shape_protor!drXrArArBmake_tensor_type_proto2      racCs2t}||_|r ||_t|||}|j||S)z8Makes a ValueInfoProto based on the data type and shape.)rr]r^rarr)r]rrPr^rRrNr^rArArBmake_tensor_value_info>s  rdc CrT)z@Makes a SparseTensor TypeProto based on the data type and shape.NrUrVz. Needs to be of int or text.)rsparse_tensor_typerrPrXrdrrMrYrrrHrZrYr[r\) rrPrRr]sparse_tensor_type_protosparse_tensor_shape_protor!r`rXrArArBmake_sparse_tensor_type_protoPrbrhcCs6t}||_|r ||_t|||}|jj|j|S)zEMakes a SparseTensor ValueInfoProto based on the data type and shape.)rr]r^rhrrer)r]rrPr^rRrNrfrArArBmake_sparse_tensor_value_info}sriinner_type_protocCt}|jj||S)zMakes a sequence TypeProto.)r sequence_typerrrjr]rArArBmake_sequence_type_protorncCrk)zMakes an optional TypeProto.)r optional_typerrrmrArArBmake_optional_type_protororq value_typecCs t}||j_|jj||S)zMakes a map TypeProto.)rmap_typerrrr)rrrr]rArArBmake_map_type_protosrtr]cCs&t}||_|r ||_|j||S)z1Makes a ValueInfoProto with the given type_proto.)rr]r^rr)r]r]r^rNrArArBmake_value_infos  rur&cCsbt|tr|}nt|tr|jddd}nt|}t|dkr!|S|dddt|ddS)Nrignore)errors@z ...<+len=>)rrYrdecoder)r& sanitizedrArArB _sanitize_strs    r|elem_shape_denotationcCs>t}||_|r ||_t|||}t|}|jj|j|S)zJMakes a Sequence[Tensors] ValueInfoProto based on the data type and shape.)rr]r^rarnrrlr)r]rrPr^r}rNr^sequence_type_protorArArBmake_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|drQ|tt|jn|drxt |j j dkrd|dnt |j j }|dt|j |dn|dr|d|jjd||jn|dr|d|jdn|dr|dn|jr||||jn|jr||||jn|jr|tttt|jn|jr|dnz|jr|d nq|jr|d!t|jD]\} } | t |jd"krd#nd$} |d| d| q|d%n?|jrS|d!t|jD]\} } | t |jd"kr7d#nd$} |d| jd| q'|d%||jn|d&|rbd'||fSd'|S)+N=rr$r7rYcSs|dS)Nz.15grA)rrArArB str_floatsz&printable_attribute..str_floatr!rHcSst|SrQ)rY)r!rArArBstr_intsz$printable_attribute..str_int_Tstr_elemCallable[[_T], str]xs Sequence[_T]cSsddt||dS)N[, ])joinmap)rrrArArBstr_listsz%printable_attribute..str_listr&rrzzz[, ...]z[, ...]rr,rcrz  )rr$r7rY)r!rHr7rY)rrrrr7rY)appendr]rHasFieldrr!reprr|r&rrrrrrr(r*r9r8r:rYr/rr;r<r>rYr=rdr) rArcontentrrrrr=rr!r*commar(rArArBprintable_attributesj                       rrXTensorShapeProto.DimensioncCs$|d}|dur dStt||S)Nrb?) WhichOneofrYr)rXwhichrArArB printable_dim/s rrcCs|ddkr5tj|jj}|jdr3t|jjj r/|t dd t t |jjj 7}|S|d7}|S|ddur>dSd|dS) NrbrWrPrrT, scalarrcz Unknown type )rrDataTypeNamerWrrrrPrXrYrrrrr&rArArBprintable_type6s "rrcCs,d|j}|jr|dt|jd}|S)N%rr)r]rr)rr&rArArBprintable_value_infoDs rcCshd|jd}|tj|j7}|jdur.t|jr*|tddt t|j7}n|d7}|d7}|S)NrrrrTrr) r]rrrrrrrYrrrrArArBprintable_tensor_protoKs   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|qSrrArSr]rArArBrK\z"printable_node..rrz1printed_attr_subgraphs[1] must be an instance of rrzprinted must be an instance of cSrrrArrArArBrKprrz](r(r)rrerrrfrrr/rrdrYrgrcrX) rirrrr= printed_attrsrAprinted_attr_subgraphsprintedprinted_attributesprinted_inputsrArArBprintable_nodeWs6        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 rycSh|]}|jqSrArIrrArArBrz"printable_graph..rrrz,optional inputs with matching initializers (cSrrArIrSr!rArArBrrcsg|] }|jvrt|qSrA)r]rr graph_inputsrArBrKs  z#printable_graph..zinitializers ({T)rrz-contents_subgraphs[1] must be an instance of rrr7rcSsg|]}d|jqSrrI)rSoutrArArBrKs} )r]rsrrcrrrrirrr/rrdreprintable_graph)ryrrindentheader initializersin_strsin_with_init_strsinpline init_strsr=ricontents_subgraphstailr(rArrBrzsj               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 rr^N)rgoogleprotobufmessageMessager DESCRIPTORfieldsr] ClearFieldr TYPE_MESSAGElabelLABEL_REPEATEDrstrip_doc_stringr)r descriptorrTrArArBrs       r algorithmalgorithm_bindingsAssignmentBindingTypeinitializationGraphProto | Noneinitialization_bindingsAssignmentBindingType | NonercCsrt}|j||D]\}}|j}||_||_q |r#|j||r7|D]\}}|j}||_||_q'|SrQ) rrrupdate_bindingrrNrbrinitialization_binding)rrrr training_inforrbindingrArArBmake_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_MAPrrrArArBr 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_dtyperrArArB$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 )rrr]rrArArBtensor_dtype_to_stringrrcCsZttjdttjdttjdttjdttjdttjdttjdi}|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) rHrr%rrDOUBLErrrrrr)rstorage_tensor_type_to_fieldrArArBrs       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|qSrA)r)rSrrrArArB =s z,np_dtype_to_tensor_dtype..rzUnable to convert type z into TensorProto element type.) rrrhtypingcastr issubdtypestr_rrrM)r_np_dtype_to_tensor_dtyperArArBnp_dtype_to_tensor_dtype3s  r KeysView[int]cCs tjS)zcGet all tensor types from TensorProto. Returns: all tensor types from TensorProto )rrrrArArArBget_all_tensor_dtypesJs rcCsi|]\}}||qSrArA)rSrrrArArBrSs rzdict[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)rrArArBr@Xs r@)r6r4r7r8)F)rDrErFrGr7rH)NNNN)rXrYrZr[r\r[r]rJr^rJrIrJr_rJr`rr7r)rIrYrKrHr7r)rprqr]rYrZrrr\rrrsrtr^rJrurvrwrxr7r )NNNNN)rIrYr}rYrZr[r\r[rprqr~rErrrrr^rJr_rJrurvr7r )ryr r`rr7r)rrrrr7r)rrrrr7r) r]rYrrHrrrrrrGr7r)rrrrrrr7r)r]rYrrrrr7r ) r]rYrrHrrrr r7r)r]rYrrrbr r7r )rbrr7r)NN) rNrYrbrr^rJrrr7r rQ) r]rYrrEr^rJrDrJr7r )rAr r7r)rirrHrYr7r)r]rYr7r)rrHrPrQrRrSr7r)rcN) r]rYrrHrPrQr^rYrRrSr7r)rjrr7r)rrHrrrr7r)rc)r]rYr]rr^rYr7r)r&rr7rY) r]rYrrHrPrQr^rYr}rSr7r)rAr rrGr7r)rXrr7rY)rrr7rY)rrr7rY)rrr7rY)rcF)rirrrYrrGr7r)ryr rrYr7rY)rrr7r) rr rrrrrrr7r)rrHr7r)rrHr7rH)rrHr7rY)rrr7r)r7r)rrHr7rY)i __future__rcollections.abcr, functoolsrrrrrrgoogle.protobuf.messagernumpyrtyping_extensionsrrronnx.onnx_data_pbrr r onnx.onnx_pbr r r rrrrrrrrrrr#google.protobuf.internal.containersrtuplerYrHVersionRowTyper/r4rr5__annotations__dictr8rCrLrWrjrorzr|rrrrrrrrrrrrarFrGrMrOrardrhrirnrqrtrur|rrrrrrr deprecatedrrrrrr lru_cacherrrrrhrr@rArArArBs4 "$    1 + $ %  {      a " $  1 1    P    # P