bIgUddlmZddlZddlZddlZddlmZddlm Z m Z m Z m Z m Z mZmZmZmZmZddlZddlZddlmZddlmZmZmZmZmZm 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-eee.e/e/e/fee.e/e/e/e/ffZ0e e0Z1e ee.e.fZ2gdZ3de4d<e ee.e/fe/fZ5d>d Z6e6e3Z7 d? d@d Z8 dA dBd Z9 dCd Z: dA dDd Z;dCdZ< dE dFdZ=dGdZ>dGdZ? dHdZ@dIdZAdJdZBd?dKdZC dL dMdZD dN dMdZEdOdZF d? dPdZG dQdZH dRdZI dSdZJ dTdZKdUdZL dV dWdZM dX dYd ZNdZd!ZOd[d"ZPd\d#ZQ dX d]d$ZR d^ d_d%ZS dX d]d&ZT d^ d_d'ZU d`d(ZV d`d)ZW dad*ZX db dcd+ZYddd,ZZ d^ ded-Z[ d? dfd.Z\dgd/Z]dhd0Z^did1Z_djd2Z` dk dld3Zadbdmd4Zbdnd5Zc dod6Zddpd7Zedqd8Zfdrd9Zgdrd:Zhdsd;Zidtd<ZjejjDcic]\}}|| c}}Zmdud=Znycc}}w)v) annotationsN)isnan) AnyCallableDictKeysViewListSequenceTupleTypeVarUnioncast) IR_VERSIONAttributeProto FunctionProto GraphProtoMapProto ModelProto NodeProtoOperatorSetIdProto OptionalProto SequenceProtoSparseTensorProto TensorProtoTensorShapeProtoTrainingInfoProto TypeProtoValueInfoProtodefsmappingsubbyte))z1.0r#)z1.1r"r#)z1.1.2r"r#)z1.2r"r#)z1.3r"r#)z1.4.1 r#)z1.5.0r$ r#)z1.6.0r% )z1.7.0r& r,r#)z1.8.0r& r,r#)z1.8.1r&r.r,r#)z1.9.0r&r,r#)z1.10.0r'r,r#)z1.10.1r'r0r,r#)z1.10.2r'r0r,r#)z1.11.0r'r"r#)z1.12.0r'r"r#)z1.13.0r'r"r#)z1.13.1r'r3r"r#)z1.14.0r)r"r#)z1.14.1r)r4r"r#)z1.15.0r)r(r#)z1.16.0r*r$r#)z1.17.0r*r$r#VersionTableType VERSION_TABLEc0idfd }|D]}|| S)zOCreate a map from (opset-domain, opset-version) to ir-version from above table.ch~tgd|D]}|vs||<|ddk(s|d|df<!y)N)ai.onnxz ai.onnx.mlai.onnx.trainingrr=zai.onnx.preview.trainingr#)zip)release_version ir_versionargspairresults H/var/www/html/answerous/venv/lib/python3.12/site-packages/onnx/helper.pyprocessz-create_op_set_id_version_map..processZsQ EtL MD6!)t 700BLF5tAw>?  M)r?strr@intrArreturnNone)tablerErowrCs @rDcreate_op_set_id_version_maprNVs+FM  MrFcJddfd |rtfd|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) r"cT|xsd|f}|tvr t|SrStd)Nr<zUnsupported opset-version.)OP_SET_ID_VERSION_MAP ValueError)domainversionkeydefault_min_versionignore_unknowns rDfind_minz)find_min_ir_version_for..find_minys;"G, ' '(- - & &566rFc3XK|]!}|j|j#ywN)rSrT).0xrXs rD z*find_min_ir_version_for..s FQ8AHHaii0Fs'*)rS str | NonerTrHrIrH)max) opsetidlistrWrVrXs `@@rDfind_min_ir_version_forrajs,7F+FFF rFc Rt}||_|jj||jj||r||_|r||_|||_|||_|r9|jjdt|jD|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 c3>K|]\}}|t||ywrZ)make_attribute)r[rUvalues rDr]zmake_node..s) U  3 & s) rop_typeinputextendoutputname doc_stringrSoverload attributesorteditems) rfinputsoutputsrjrkrSrlkwargsnodes rD make_noderts8 ;DDLJJfKKw  $       $V\\^4  KrFc6t}||_||_|SzConstruct an OperatorSetIdProto. Args: domain (string): The domain of the operator set id version (integer): Version of operator set id Returns: OperatorSetIdProto rrSrT)rSrT operatorsetids rDmake_operatorsetidrys"'(M!M#M rFc|g}|g}|g}t}|jj|||_|jj||j j||j j||jj||jj||r||_ |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 SparseTensorProto Returns: GraphProto ) rrsrhrjrgri initializersparse_initializer value_infork) nodesrjrprqr{rkr}r|graphs rD make_graphrs0 ! LE JJeEJ KKv LL  [) ##$67 J'% LrFc6t}||_||_|Srvrw)rSrTopsetids rD make_opsetidrs!"GGNGO NrFc |g}|g}| g} t} || _|| _| jj || j j || j j || jj || jj || jj ||r|| _ | | | _ | jj | | SrZ) rrSrjrgrhrirs opset_importrmattribute_protorkrlr}) rSfnamerprqr~ opset_imports attributesattribute_protosrkrlr}fs rD make_functionrs  AAH AFGGNN6HHOOGFFMM%NN-(KKz"-.!  LL # HrFc t}t|_|jj |d}|j dd}||j j|n3|j j}tj|_ d}|j dd}||jj||jD]\}}t||||S)zConstruct a ModelProto Args: graph (GraphProto): *make_graph* returns **kwargs: any attribute to add to the returned instance Returns: ModelProto Nr functions)rrr@rCopyFrompoprrhaddronnx_opset_versionrTrrosetattr)rrrmodelrimprkvs rD make_modelr's LE"E KK9=MJJ5M  !!-0  $$&--/ 04I ;-I y) 1q! LrFc jd}||vr"d}|j|g}t|||<t|fi|S)Nr@r)getrar)rrrir_version_fieldopset_imports_fieldimportss rDmake_model_gen_versionrLsG#v%-**0"5#:7#C e &v &&rFc|jdd=|jD]-\}}|jj}||_||_/yrZ)metadata_propsrorrUre)proto dict_valuerrentrys rDset_metadata_propsrUsP Q  "1$$((*  rFct||yrZ)r)rrs rDset_model_propsrgs uj)rFctt|dzDcgc].}|dzdk(r||dzjn||dzj0c}Scc}w)Nr,r)rangelenrealimag)cais rDsplit_complex_to_pairsrksXs2w{#  UaZAFbajoo =  s3Actjtjd|d}|r|dz St |ry|dz dzdz}||zdz S)N>&++dD18 b C Nz ) 88A;Sy  ^ " N s7CL3xKWSBRxqQV}$qR"W~%Qqb>D4xaq>A%HdaiQ$(^q5Hq(3x'WSBQws qBwrQwqBw7{X1w;$J$&1HC3x "  4KC3xC3x Nz )#:  88A;Sy #:  ^ " N 63wL3xKSW81R=(C1b>)CU1HCR"W~t8!GD1H~)DQ$!)_dQh19L1HC,3x+SW73JC17NC27NC17NCd t+t Kq8|Wd d*q3x &t  3x s 3xt 3xrFc||z }tjtjdt j |d}|dzdz }|r9|r6|dzdk(ry|dzdk(r|r|d zSy|dzd z }|d z} |d kr d }t|S|dkrs|dz } | dk\r|dd| zzz}|| d| z z z}n | d kDr|dz}nd }dd| z z} | | zr|dzs#| | dz zd kDs| | zr| | dzzr| | dz zd k(r|dz }t|S|dkrK|dz } || dzz}|| dz z}| dzr\| dzs| dzrR|d zd kr|dz }t|S|s8d}t|S|dk(r| d k(r d}t|S|r|d z}t|Sd}t|S|sO|sL|dzdk(rd |zSt j |r |rd|zSd|zS|dzd z }|d z} |d k7r|dkr t|S|dkrp|dz } | dk\r|dd| zzz}|| d| z z z}n | d kDr|dz}dd| z z} | | zr|dzs#| | dz zd kDs| | zr| | dzzr| | dz zd k(r|dz }t|S|dkr^|dz } || dzz}|| dz z}| dzr[| dzs| dzrQ|d zdkr|dz }t|S|r|dz}t|S|dz}t|S|r|dz}t|S|dz}t|Std)aConvert a float32 value to a float8, e5m2 (as int). Args: fval: float to convert scale: scale, divide *fval* by *scale* before casting it fn: no infinite values uz: no negative zero saturate: if True, any value out of range included inf becomes the maximum value, otherwise, it becomes NaN. The description of operator Cast fully describes the differences. Returns: converted float rrrrrrirrrrmrpor#r7r6r,rii {|nqz%fn and uz must be both False or True.)rHrrrrrrrrs rDfloat32_to_float8e5m2r s, u A v{{4A7BA z>b C b Nz ) Nz )Tz! ^ " N s7CH3xGWSBRxqQV}$qR"W~%Qqb>D4xaq>A%HdaiQ$(^q5Hq$3x#WSB 27NC 17NC8|!g+1x<$J$&1HC3x"C3x #X!q&C 3x  4KC3xC3x  Nz )#:  88A;cz!#:  ^ " N 63wD3xCSW81R=(C1b>)CU1HCR"W~t8!GD1H~)DQ$!)_dQh19L1HC$3x#SWrQwqBwxz&pack_float32_to_4bit..sw::1aHrFN) isinstancerndarrayasarrayrravelprodshapeappendarray frompyfuncastypeuint8)rr array_flat is_odd_volume single_funcfuncarrs ` rDpack_float32_to_4bitrs eRZZ ( 5 3JGGEKK(1,1MYYz288QC=9 HK ==a +D z!$Q$ADqD!1 2C ::bhh rFc t}||_||_|tjk(r |r t dt |}d}|r|tj k(rd}nt|tjtjtjtjfvrd}n1|tjtjfvrd}n |j}t|tj ur(t#|j$dkDr|j'}|D]}||z} t#||k7rN|tjtjfvrt#||dzk(st)d|dt#|d|r ||_n|tj,tj.fvr t1|}nO|tj2k(r`tj4|j7|j9tj:j'j=}n|tj tjtjtjtjfvrtj t>tjt@tjd tjtBtjd i|} tEtG| tj4|j7|j'j=}n|tjtjfvr=|tjk(} tI|| j'j=}nw|tjJk(r)tj4|j7tL}n;|tjk(r(tj4|j7tN}tQ|} tS|| jU||jVjU||S) aMake 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 (string): tensor name data_type (int): a value such as onnx.TensorProto.FLOAT dims (List[int]): shape vals: values raw (bool): 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.r#r,g?z8Number of values does not match tensor's size. Expected z , but it is z. rct|ddiS)NrT)rrAs rDrzmake_tensor..s:O;";rFct|dddS)NT)rr)rrs rDrzmake_tensor..s:Odt;rF)r),r data_typerjSTRING TypeErrortensor_dtype_to_np_dtypeBFLOAT16 FLOAT8E4M3FNFLOAT8E4M3FNUZ FLOAT8E5M2FLOAT8E5M2FNUZUINT4INT4itemsizetyperrrrflattenrRraw_data COMPLEX64 COMPLEX128rFLOAT16rrviewuint16tolistrrrlistmaprBOOLrHbytestensor_dtype_to_fieldgetattrrhdims) rjrr"valsrawtensornp_dtype expected_sizedfcastrfields rD make_tensorr+s(]F FFKK&&&3DEE' 2HM  ,, ,M   $ $  & &  " "  & &   M ;,,k.>.>? ?M$--M DzRZZC Oa$7||~   4yM! +++[-=-=> >D ]S00J=/Yefijnfoepprs   .. 0F0FG G)$/D +-- -%%h/442994EMMOVVX    $ $  & &  " "  & &   $$&9((*?**-&&(=**-  EHHTN))(3;;=DDFD         +"2"22F(V<DDFMMOD +** *88D>((-D +,, ,88D>((/D%i0%%d+ KKt MrFct}|jj||jj||jj ||S)zConstruct a SparseTensorProto Args: values (TensorProto): the values indices (TensorProto): the indices dims: the shape Returns: SparseTensorProto )rvaluesrindicesr"rh)r-r.r"sparses rDmake_sparse_tensorr0sG F MM6" NNG$ KKt MrFct}||_||_|tjk(r|S|tjk(r |j }n|tj k(r |j}nk|tjk(r |j}nK|tjk(r |j}n+|tjk(r |j}n td|j!||S)z/Make a Sequence with specified value arguments.z8The element type in the input sequence is not supported.)rrj elem_type UNDEFINEDTENSOR tensor_values SPARSE_TENSORsparse_tensor_valuesSEQUENCEsequence_valuesMAP map_valuesrOPTIONALoptional_valuesr rh)rjr2r-sequencerms rD make_sequencer?/s HHM"HM+++M(((** m11 111 m,, ,,, m'' ''' m,, ,,, RSS V OrFct}tjtjtjtj tj tjtjtjg}||_ ||_ |tjk(r|jj|n||vr|jj||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 )rrINT8INT16INT32INT64UINT8UINT16UINT32UINT64rjkey_typer string_keysrhkeysr-r)rjrIrKr- map_protovalid_key_int_typess rDmake_maprNLs I IN!I;%%%$$T* ( (d# f% rFct}||_||_|tjk(r|S|tjk(r |j }n|tj k(r |j}nk|tjk(r |j}nK|tjk(r |j}n+|tjk(r |j}n td|j||S)z0Make an Optional with specified value arguments.z8The element type in the input optional is not supported.)rrjr2r3r4 tensor_valuer6sparse_tensor_valuer8sequence_valuer: map_valuer<optional_valuer r)rjr2reoptionalrms rD make_optionalrVks HHM"HM+++M((()) m11 100 m,, ,++ m'' '&& m,, ,++ RSS u OrFcHt|tr|S|jdS)z2Coerce a string (or bytes) value into UTF-8 bytes.utf-8)rrencode)res rD _to_bytesrZsue,5G%,,w2GGrFc  t}||_|r||_t|tj r't ||_tj|_ nt|tjr't||_ tj|_ nt|ttfr't!||_tj$|_ nLt|t&r2|j(j+|tj,|_ n t|t.r2|j0j+|tj2|_ nt|t4r2|j6j+|tj8|_ nt|t:r2|j<j+|tj>|_ nDt|t@jBjDrtG|}tI|dk(r|tKd|d||Dchc] }t|}}tj tjLftjtjNfttftjPft&tjRft.tjTft4tjVft:tjXffD]\}t[fd|Ds|}n| tKd|tjLk(r2|j\j_|tjL|_ n|tjNk(r2|j`j_|tjN|_ nu|tjPk(r9|jbj_d|DtjP|_ n)|tjRk(r1|jdj_|tjR|_ n|tjTk(r1|jfj_|tjT|_ n|tjVk(r1|jhj_|tjV|_ n]|tjXk(r1|jjj_|tjX|_ ntmtod|d|M|j|k7r>tod tq|jd |jd tq|d |d |Scc}w) z0Makes an AttributeProto based on the value type.rzCould not infer attribute `z` type from empty iteratorc36K|]}t|ywrZ) issubclass)r[texp_ts rDr]z!make_attribute..s;z!U+;szRCould not infer the attribute type from the elements of the passed Iterable value.c32K|]}t|ywrZ)rZ)r[rs rDr]z!make_attribute..s< ! #6#67~445u~556n445"N$A$AB^223N667$ x;U;; (I   h ++ + II  U #&++DI .// / KK  u %&--DI .00 0 LL  k(rt'|j@S|jtjBk(rytd|)Nz)Cannot get value of reference attribute: zUnsupported ONNX attribute: )" ref_attr_namerRrrrhrrerrrir4r^r6rjrlrkrnrmrsrr{rrrzrtr|rur}rvr~rwrrxrr3)rs rDget_attribute_valuers DTFKLL yyN(((vv  yyN&&&vv  yyN)))vv  yyN)))vv  yyN000!!! yyN(((vv  yyN---ww yyN)))DKK   yyN'''DII yyN***DLL!! yyN***DLL!! yyN111D''(( yyN)))DKK   yyN...D$$%% yyN,,, 3D6: ;;rFc|jDcgc]}|j|k(s|}}t|dkDrtd|t|dkrtd|t |dScc}w)Nr#z'Node has multiple attributes with name z Node has no attribute with name r)rmrjrrRr)rs attr_namer\matchings rDget_node_attr_valuersw>>AaQVVy-@AHA 8}qB9+NOO 8}q;I;GHH x{ ++ Bs A0A0c(t}||_|SrZ)rrj)rjvalue_info_protos rDmake_empty_tensor_value_infor$s%'  rFct}|j}||_|j}||jj g|r"t |t |k7r tdt|D]n\}}|jj}|n?t|tr||_ n't|tr||_ntd|d|se|||_p|S)z:Makes a Tensor TypeProto based on the data type and shape.>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_typer2rdimrhrrR enumeraterrrH dim_valuerG dim_param denotation) r2rshape_denotation type_prototensor_type_prototensor_shape_protorr(rs rDmake_tensor_type_protor*s J".."+*00  %%b) $4 5U CP e$ 5DAq$((,,.CyAs# ! As# !  -aS0LM !1!!4 5 rFct}||_|r||_t|||}|jj ||S)z8Makes a ValueInfoProto based on the data type and shape.)rrjrkrrr)rjr2rrkrrrs rDmake_tensor_value_inforWsM&' &0#.y%AQR""#45 rFct}|j}||_|j}||jj g|r"t |t |k7r tdt|D]n\}}|jj}|n?t|tr||_ n't|tr||_ntd|d|se|||_p|S)z@Makes a SparseTensor TypeProto based on the data type and shape.rrz. Needs to be of int or text.)rsparse_tensor_typer2rrrhrrRrrrrHrrGrr) r2rrrsparse_tensor_type_protosparse_tensor_shape_protorr(rs rDmake_sparse_tensor_type_protoris J)<<)2& 8 > >  "%%,,R0 $4 5U CP e$ 5DAq+//335CyAs# ! As# !  -aS0MN !1!!4 5 rFct}||_|r||_t|||}|jj j |j |S)zEMakes a SparseTensor ValueInfoProto based on the data type and shape.)rrjrkrrrr)rjr2rrkrrrs rDmake_sparse_tensor_value_inforsb&' &0#<5* ,,55 33 rFcdt}|jjj||S)zMakes a sequence TypeProto.)r sequence_typer2rinner_type_protors rDmake_sequence_type_protor,J&&//0@A rFcdt}|jjj||S)zMakes an optional TypeProto.)r optional_typer2rrs rDmake_optional_type_protorrrFct}||j_|jjj ||S)zMakes a map TypeProto.)rmap_typerI value_typer)rIrrs rDmake_map_type_protors9 J#+J ""++J7 rFcpt}||_|r||_|jj ||S)z1Makes a ValueInfoProto with the given type_proto.)rrjrkrr)rjrrkrs rDmake_value_infors< &' &0#"":. rFct|tr|}n/t|tr|jdd}n t|}t |dkr|S|dddt |dz dzS)NrXignore)errors@z ...<+len=>)rrGrdecoder)ri sanitizeds rD _sanitize_strrsl!S Au HHWXH6 F  9~ Sb>iY")<'>a@ @@rFct}||_|r||_t|||}t |}|j j j|j |S)zJMakes a Sequence[Tensors] ValueInfoProto based on the data type and shape.)rrjrkrrrrr)rjr2rrkelem_shape_denotationrrsequence_type_protos rDmake_tensor_sequence_value_inforse&' &0#.y%AVW23DE''001D1R1RS rFc bg}|j|j|jddd}dd}td}dd}g}|jdr#|j||jn|jdr#|j||j n[|jdr/|jt t|jn|jd rt|jjd kDr|jd nt|jj}|jd t|j|d n|jdrF|jd|jjd |j|jn4|jdr!|jd|j d n|j"r$|j|||j"n|j$r$|j|||j$n|j&r=|jt)t+t-t|j&nY|j.r|jdn:|j0rv|jdt3|j0D];\} } | t|j0dz k7rdnd} |jd| d | =|jdn|j4r|jdt3|j4D]E\} } | t|j4dz k7rdnd} |jd| jd | G|jd|j7|j4n|jd|rdj9||fSdj9|S)N=rc |dS)Nz.15grK)rs rD str_floatz&printable_attribute..str_floatsDrFrct|SrZ)rG)rs rDstr_intz$printable_attribute..str_ints 1v rF_TcDddjt||zdzS)N[, ])joinr)str_elemxss rDstr_listz%printable_attribute..str_list s"TYYs8R011C77rFrir^rzz, ...]rr#,rz  )rrgrIrG)rrHrIrG)rzCallable[[_T], str]rz Sequence[_T]rIrG)rrjr HasFieldrrreprrrirr^r"r rr!rkrmr{rzr|rGrrr}rrrrhr) r subgraphscontentrrrrrr*rrmcommarks rDprintable_attributersG NN499 NN3  B8F }}Sy() s wtvv' s tM$&&123 s  tvv{{ a  NN: &*$&&*:*:;E NN_WTVVU-C,DAF G s  Q/0 dff t dggYa01 x 4;;78 x34 s4M4<< @ABC ()  st//0 8EArD$4$4 5 99CrE NN\"Qug6 7 8 s sdkk* 7DAqDKK 01 44C"E NNWQVVHAeW5 6 7 s dkk"{#xx &(( 88G rFcT|jd}|ytt||S)Nre?) WhichOneofrGr!)rwhichs rD printable_dimrDs* NN7 #E } wsE" ##rFc |jddk(rtjj|jj }|jj dr|t|jjjrN|tddjtt|jjjzz }|S|dz }|S|jdyd|jdS) Nrerrrr\, scalarrz Unknown type )rrDataTypeNamerr2rrrrrGrrrr^ris rDprintable_typerKs||G -  % %amm&=&= > == ! !' *1==&&**+S]AMM!&&)*! , HrFc Fd|jd}|tjj|jz }|j St |j r9|tddjtt|j zz }n|dz }|dz }|S)Nrrrr\rr) rjrrrrr"rrGrrrs rDprintable_tensor_protor`s AFF81 A   " "1;; //Avv qvv; TCHHSaff%5667 7A OAHA HrFc g}t|jrL|jdj|jDcgc]}d| c}|jdg}g}|jD]}|r[t ||}t |dtstdtd|j|d|j|d`t |} t | tstdtd|j| djt|} dj|jDcgc]}d| c}} |jr%|j|jd | d | d n!|j|jd | d |r|d j|z|fS|d j|zScc}wcc}w)Nrrrr#z1printed_attr_subgraphs[1] must be an instance of .rzprinted must be an instance of rz](rb(r)rrirrrmrrrr rhrGrnrgrf) rsprefixrrrjr printed_attrsrprinted_attr_subgraphsprintedprinted_attributesprinted_inputss rDprintable_noderlsG 4;;tyy!EAdV*!EFGs!FM * %8y%I "4Q7>GvQO MM03 4  !7!: ;)$/Ggs+"A#a HII   ) *6-#89YYtzzBt!D6 BCN ~~$,,q);(NaPQ$,,q(8:;))611 CHHW% %%7"F( Cs G  Gc g}|dz}d|jg}|jDchc]}|j}}t|jr|j dg}g}|jD]E} | j|vr|j t | ,|j t | G|rC|j |dj |zg}|D]} |j |dz| z|j d|re|j d|j |dj |zg}|D]} |j |dz| z|j dt|t|kr|jD chc]} | j} } |jD cgc]} | j| vr t| } } |j d|j |dj |zg}| D]} |j |dz| z|j d|j d|j |dj |zg}|jD]^}t||d }t|d tstd td |j |d|j|d `dg}t|jrE|j dj |jDcgc]}d|jc}|j |dj |z|j |dz|D]}|j dt|z!dj |Scc}wcc} wcc} wcc}w)zDisplay a GraphProto as a string. Args: graph (GraphProto): the graph to display prefix (string): prefix of every line Returns: string z rrrrbz,optional inputs with matching initializers (zinitializers ({T)rr#z-contents_subgraphs[1] must be an instance of rrrIrr} )rjr{rrgrrrrrsrrrr rhriprintable_graph)rrrindentheaderr^ initializersin_strsin_with_init_strsinpliner graph_inputs init_strsrrscontents_subgraphstailoutrks rDrrsoG d]Fuzz "F$)$5$56qAFF6L6 5;; c  ;; DCxx|+3C89!(()=c)BC  D  NN6CHHV$44 5F 5v}t34 5 c  MMH I NN6CHHV$44 5F) 5v}t34 5 MM#   !C $5 5,1KK8qAFF8L8**66-'q)I MM* + NN6CHHV$44 5F! 5v}t34 5 MM#  MM# NN6CHHV,,-!F -+D&DI,Q/6KD6QRST T)!,- (+, - :D 5<< DIIF#388*~FGH NN6CHHTN*+ NN6C< 2toa0012 99W }7<92GsN0N5= N:#N? cxt|tjjjs1t dtjjjd|j jD]}|jdk(r|j|j.|j|jk(sH|j|jk(r't||jD] }t||j!|jstt||jy)z:Empties `doc_string` field on any nested protobuf messageszproto must be an instance of rrkN)rgoogleprotobufmessageMessager  DESCRIPTORfieldsrj ClearFieldr TYPE_MESSAGElabelLABEL_REPEATEDr!strip_doc_stringr)r descriptorr\s rDrrs eV__44<< =+FOO,C,C,K,K+LA N  &&--B ??l *   Z__ - __ 7 7 7:#<#<<  8(A$Q'( 0  !@ABrFcVt}|jj||D]-\}}|jj }||_||_/|r|jj||r2|D]-\}}|jj }||_||_/|SrZ) r algorithmrupdate_bindingrrUreinitializationinitialization_binding)ralgorithm_bindingsrinitialization_bindings training_inforrbindings rDmake_training_infors &'M$$Y/"1..224   $$--n=+ DAq#::>>@GGKGM  rFc<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 )r TENSOR_TYPE_MAPr& tensor_dtypes rDr r s  " "< 0 9 99rFc<tj|jS)zConvert a TensorProto's data_type to corresponding data_type for storage. Args: tensor_dtype: TensorProto's data_type Returns: data_type for storage )r r  storage_dtyper!s rD$tensor_dtype_to_storage_tensor_dtyper%s  " "< 0 > >>rFc<tj|jS)zGet the name of given TensorProto's data_type. Args: tensor_dtype: TensorProto's data_type Returns: the name of data_type )r r rjr!s rDtensor_dtype_to_stringr' s  " "< 0 5 55rFc^tjtj|jS)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 )r _STORAGE_TENSOR_TYPE_TO_FIELDr r$r!s rDr r ,s+  0 0 -;; rFc|tjvr!tttj|St j |tj rtjS|tjtjtjtjtjtjtj hvr#tj"|j$ddSt'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 rzUnable to convert type z into TensorProto element type.)r _NP_TYPE_TO_TENSOR_TYPErrHr issubdtypestr_rrcustom_np_typesbfloat16 float8e4m3fnfloat8e4m3fnuz float8e5m2float8e5m2fnuzint4uint4mapping_name_to_data_typedescrrR)r&s rDnp_dtype_to_tensor_dtyper8:s7222   + +H 5   }}Xrww'!!!  $$&&""&&889J19MNN  !(-LM rFc>tjjS)zcGet all tensor types from TensorProto. Returns: all tensor types from TensorProto )r r rKrKrFrDget_all_tensor_dtypesr:\s  " " ' ' ))rFc|tjjvr t|Stjj dS)zConvert AttributeProto type to string. Args: attr_type: AttributeProto type. Returns: String representing the supplied attr_type. r)r AttributeTyper-_ATTRIBUTE_TYPE_TO_STRrK)rs rDrrhs@N007799%i00  ' ' , , .q 11rF)rLr8rIVersionMapType)F)r`Sequence[OperatorSetIdProto]rWboolrIrH)NNNN)rfrGrp Sequence[str]rqrArjr^rkr^rSr^rlr^rrrrIr)rSrGrTrHrIr)r~Sequence[NodeProto]rjrGrpSequence[ValueInfoProto]rqrCr{zSequence[TensorProto] | Nonerkr^r}Sequence[ValueInfoProto] | Noner|z"Sequence[SparseTensorProto] | NonerIr)NNNNN)rSrGrrGrprArqrAr~rBrr?rzSequence[str] | NonerzSequence[AttributeProto] | Nonerkr^rlr^r}rDrIr)rrrrrrIr)rzRModelProto | GraphProto | FunctionProto | NodeProto | TensorProto | ValueInfoProtordict[str, str]rIrJ)rrrrErIrJ)rzSequence[np.complex64]rI Sequence[int])rrgrr@rIrH)?TFT) rrgrrgrr@rr@rr@rIrH)rGFFT)rznp.ndarray | Sequencerr@rIz np.ndarray) rjrGrrHr"rFr#rr$r@rIr)r-rr.rr"rFrIr)rjrGr2zSequenceProto.DataTyper-z Sequence[Any]rIr) rjrGrIrHrKz list[Any]r-rrIr)rjrGr2zOptionalProto.DataTyperez Any | NonerIr)re str | bytesrIr)NN) rUrGrerrkr^rz int | NonerIrrZ)rjrGrzAttributeProto.AttributeTyperkr^rIr)rrrIr)rsrrrGrIr)rjrGrIr)r2rHr!Sequence[str | int | None] | Nonerlist[str] | NonerIr)rN) rjrGr2rHrrIrkrGrrJrIr)rrrIr)rIrHrrrIr)r)rjrGrrrkrGrIr)rirHrIrG) rjrGr2rHrrIrkrGrrJrIr)rrrr@rI"str | tuple[str, list[GraphProto]])rzTensorShapeProto.DimensionrIrG)r^rrIrG)rrrIrG)r^rrIrG)rF)rsrrrGrr@rIrK)rrrrGrIrG)rzgoogle.protobuf.message.MessagerIrJ) rrrAssignmentBindingTyperzGraphProto | NonerzAssignmentBindingType | NonerIr)r"rHrInp.dtype)r"rHrIrH)r"rHrIrG)r&rMrIrH)rIz KeysView[int])rrHrIrG)o __future__rcollections.abcrorcrcmathrtypingrrrrr r r r r rgoogle.protobuf.messager numpyronnx._custom_element_types_custom_element_typesr.onnxrrrrrrrrrrrrrrrrrr r!rGrHVersionRowTyper8rLr9__annotations__r>rNrQrartryrrrrrrrrrrrrr+r0r?rNrVrZrdrrrrrrrrrrrrrrrrrrrrrrrr r%r'r r8r:r<ror=r)rrs00rDrYs|#    4,uS#sC/0%S#sC8O2PPQ'U38_-# 8eCHos*+"5]CGL-?C@!. . ..  .  .  ....b  .15!26=A( ( ( %(& ( . (  (0(;((V ,(,8<!26! ! !  !  !  ! 0 ! %! 6! ! ! 0! ! H J'  $* " C C C C C  C  CP {K {K {K {K {K  {K  {K| 6LQu uu%2u:=uDHuup "-5B( %  : $-7D > %  :H" [ [ [[ [  [~RV   6 DN  !