h?gSSKJr SSKrSSKJrJrJr SSKrSSK J r SSK J r J r Jr SSKJrJrJr SSKJr SSKJrJrJrJr SS KJrJrJrJrJr S r "S S \!5r""S S\!5r#"SS\$5r%"SS5r&"SS5r'S&Sjr(\("5r)"SS5r*"SS5r+S'Sjr,S(Sjr-"SS5r."SS\R^5r0"S S!\05r1"S"S#\05r2"S$S%\25r3g))) annotationsN)AnyClassVarIterable) TensorProto)get_all_schemas_with_history get_schemaonnx_opset_version) make_nodemake_tensor_type_protonp_dtype_to_tensor_dtype)to_array)AttributeProto GraphProto NodeProto TypeProto)bfloat16 float8e4m3fnfloat8e4m3fnuz float8e5m2float8e5m2fnuzcSU;a!URSSS9up[U5nX4$US4$![a US4s$f=f)N_)maxsplit)rsplitint ValueError)nameprefixversvs O/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/reference/op_run.py_split_class_namer$sZ d{{{3{3  D Ay : :  s ,==c\rSrSrSrSrg)RuntimeTypeError$z1 Raised when a type of a variable is unexpected. N__name__ __module__ __qualname____firstlineno____doc____static_attributes__r(r#r&r&$r0r&c\rSrSrSrSrg)RuntimeContextError*zh Raised when the context is missing but an context dependent implementation is defined for an operator. r(Nr)r(r0r#r3r3*r1r0r3c\rSrSrSrSrg)RuntimeImplementationError0z: Raised when no implementation was found for an operator. r(Nr)r(r0r#r6r60r1r0r6c\rSrSrSrSrg) DefaultNone6zl Default value for parameters when the parameter is not set but the operator has a default behavior for it. r(Nr)r(r0r#r9r96sr0r9c,\rSrSrSrSSjrSSjrSrg) RefAttrName=zo Implements a link between a parameter of a function and an attribute in node. :param name: name of the input cXlgNr)selfrs r#__init__RefAttrName.__init__Es r0cPURRSUR<S3$)usual()) __class__r*rrAs r#__repr__RefAttrName.__repr__Hs$..))*!DII=::r0r@N)rstrreturnrL)r*r+r,r-r.rBrJr/r(r0r#r<r<=s;r0r<c 0n[5HnURU;aURXRR:waB[SUR<SUR<SXRR<S35eURXRR:aXUR'OXUR'XURS-[ UR5-'M U$)Nz7This function assumes every operator has a unique name z even accross multiple domains z and .r)rrdomainNotImplementedError since_versionrL)resschemas r#_build_schemasrVMsC.0 ;;# }}KK 0 7 77)Mfkk_]55;]]4EU3{{K[KbKbJeefh##c++&6&D&DD#)FKK % =CFKK# F$8$8 9 9:1 Jr0c(\rSrSrSSjrSSjrSrg)OnnxTypeccj[U[5(d[S[U5S35eXlg)Nz type_proto z must be of type TypeProto.) isinstancer TypeErrortype type_proto)rAr^s r#rBOnnxType.__init__ds/*i00k$z*:);;VWX X$r0c$SUR<S3$)Nz OnnxType(rGr^rIs r#rJOnnxType.__repr__is4??-Q//r0raN)r^rrM)r*r+r,r-rBrJr/r(r0r#rXrXcs % 0r0rXcF\rSrSrSrSSjr\SSj5rSrg) SparseTensormz\ Simple representation of a sparse tensor. It is based on numpy but does not require scipy. c(XlX lX0lgr?)valuesindicesshape)rArgrhris r#rBSparseTensor.__init__ss   r0c.URR$r?)rgdtyperIs r#rlSparseTensor.dtypezs{{   r0)rhrirgN)rg np.ndarrayrhrnriz tuple[int]rNNone)rNr) r*r+r,r-r.rBpropertyrlr/r(r0r#rdrdms@  +5>H !!r0rdc[SUR55n[[UR5[UR 5U5$)z Hosts a sparse tensor. c3$# UHov M g7fr?r().0ds r# #to_sparse_tensor..s&X!Xs)tupledimsrdrrgrh)attris r#to_sparse_tensorrzs9 &SXX& &E ,hs{{.CU KKr0cURnU[R:XaqURn[ UR 5n[ R"U[S9R5n[U5H upVXdU'M URU5$U[R[R[R[R4;a[R[ [R["[R[$[R[&0nUR)S5(a UR*nO URn[ UR 5n[ R"X7US9R5n[U5H upVXdU'M URU5$[-U5$)zp Similar to :func:`to_array` but deals with bfloat16, float8e4m3fn, float8e4m3fnuz, float8e5m2, float8e5m2fnuz. )rlraw_data) data_typerBFLOAT16 int32_datarwrxnpemptyrravel enumeratereshape FLOAT8E4M3FNFLOAT8E4M3FNUZ FLOAT8E5M2FLOAT8E5M2FNUZrrrrHasFieldr|r)tensor elem_typedatariyirtms r#to_array_extendedrsZ   IK(((  fkk" HHU( + 1 1 3dODAaD$yy  """"   $ $l  & &  " "J  & &   ??: & &??D$$Dfkk" HHUI, / 5 5 7dODAaD$yy F r0c"\rSrSrSrSSjrSrg)GraphgcXlgr?r)rArs r#rBGraph.__init__sr0N)rrrNro)r*r+r,r- __slots__rBr/r(r0r#rrs Ir0rc2\rSrSr%SrSr\RS\RS\RS\RS\RS\RS \RS \RS \RS \R S \R"S\R$S\R&S\R(S0rS\S'S+S,SjjrSrS+S-Sjjr\S5rS.Sjr\S/Sj5r\S0Sj5r\S0Sj5r \S1Sj5r!\S1Sj5r"S2Sjr#S1S jr$\%RLS!5r'SSS".S#jr(\)S$5r*\)S3S4S%jj5r+\)S5S6S'jj5r,\)SS&S(.S7S)jj5r-S*r.g)8OpRuna Ancestor to all operators in this subfolder. :param onnx_node: `onnx` node :param run_params: additional parameters such as `verbose`, `opsets` (it can be more than one if the operator has a subgraph), `log` for a logging function :param schema: operator schema cB[R"UR5$r?)rfloat32frys r#OpRun.s"**SUU*;r0cnURVs/sHn[R"U5PM sn$s snfr?)floatsrr)ryrs r#rrs"3::+N:aBJJqM:+N+Ns 2c,[UR5$r?)rrrs r#rrs %,r0cXURVs/sHn[U5PM sn$s snfr?)graphsr)ryrs r#rrscjj+IjE!Hj+I+I'c,[UR5$r?)rrrs r#rrs CEE r0cXURVs/sHn[U5PM sn$s snfr?)intsr)ryrs r#rrs#(()C(Q#a&()C)Crc,[UR5$r?)rz sparse_tensorrs r#rrs2B3CTCT2Ur0cXURVs/sHn[U5PM sn$s snfr?)sparse_tensorsrzryts r#rrs+),););4 );A Q );4 4 rc8URRS5$Nzutf-8)sdecoders r#rrs355<<+@r0cbURVs/sHoRS5PM sn$s snfr)stringsr)ryrs r#rrs! ,T 1XXg-> ,T,Ts,c,[UR5$r?)rrrs r#rrs+q-A;,W,Wrc,[UR5$r?)rXtprs r#rrs x/?r0cXURVs/sHn[U5PM sn$s snfr?) type_protosrXrs r#rrscoo0Vo!o0V0VrzClassVar[dict[Any, Any]]_attribute_conversion_functionsNc [U[5(d[S[U5S35eSH$nXB;dM [ SU<S[ U5S35e SU;a [ S5eXlX lUc[URS 5(aURRUl O|URR[;a#[URRUl O;UR[;a[URUl OSUl OX0l S UlUR#5 g) Nz$run_params must be a dictionary not rP)opsetsnew_ops Attribute z must be in run_params, only z was found.logz#run_params must contains key 'log'. op_schemaF)r[dictr\r] RuntimeErrorsortedKeyError onnx_node run_paramshasattrrHr_schemar*_schemasop_type has_subgraph_load_attributes)rArrrUrys r#rBOpRun.__init__s*d++B4 CSBTTUVW W(C$" 'Dj)*+7)  "@A A"$ >t~~{33#~~77 ((H4'(?(?@ ""h.' (9(9: # !L! r0c0URS"U/UQ76 g)Nr)r)rApatternargss r#_log OpRun._logs w..r0cFUR[R:XaSSKJn UR R SS5nU"URUR S[SUR R SS5S- 5UcSS9$[UR55S9$UR[R;a#[RUR"U5$Uc[[S UR<S UR<S URR <S URR"<S US3 5e[SUR<S UR<S URR <S URR"<S US US3 5e)z2 Converts an attribute value into a python value. r)ReferenceEvaluatorrNrverbose)rrrzUnable to convert attribute z type z from node type z , domain  rPz$Unable to convert default value for )r]rGRAPH"onnx.reference.reference_evaluatorrrgetrmaxlistrgrrAttributeErrorrrrrQ)rAryref_attrrs r#_extract_attribute_valueOpRun._extract_attribute_values 88~++ + oo)))T:G%x0At229a@1DE '  6:'..:J5K   88u<< <88B3G G ? .sxxl& M""&.."8"8!;<..//2"SE<  27<<2B& U"nn4478nn++.bRy C  r0c6URSU=(d 0US9$)N attributes)runcontextvaluers r#_evaluate_subgraphOpRun._evaluate_subgraphsyyw}"yDDr0cSUl/nURRHnURnUR(a[ UR5nSUlOUR U5n[XU5 URU5 UR[R:XdMSUl U=RUR-sl[USUR3US4Sj5 M UR(Ga;URRS;Ga URRR!5HupV[#X5(aM[%USS5(a(['SU<S URR<S 35e[#US 5(dMdUR(RS :Xd>UR(RS :Xa'UR(R*R,S :XaSnOUR UR(U5n[XU5 URU5 M [/U5Ulg)zChecks and loads attributes.FT_run_Nc.[RXU5$r?)rrrs r#r(OpRun._load_attributes..,s%BZBZ Cr0>Constantrequiredrz3 is expected based on ONNX specifications for node rP default_valuer)has_linked_attributer attributer ref_attr_namer<rsetattrappendr]rrrrrritemsrgetattrrrrr}setattributes_names_)rAadded_attributesryrrkr"s r#rOpRun._load_attributess$)!>>++C88D  #C$5$56,0)55c: D &  # #D )xx>///$(!))U-G-GG)CHH:&*/D,( <<(>'ADq/22??//14OO00A5 ! 1 1 ; ;q @%)E$($A$A!//ST$UE/(//2#8$"%%5!6r0ct[U[5(d[S[U5<S35e[ 5n[ 5nUR HnUR UR5 M URHnUR UR5 M URHnUR UR5 M URHRnURHnUR U5 M URHnX;dM UR U5 M MT [U5$)z Returns all varibles not registered as inputs and not produced by an node inside the graph. This inputs are part of the context existing in the graph calling this one. zUnexpected type rP) r[rr\r]r initializeraddrsparse_initializerinputnodeoutputr) graphlocalknowninit sparse_initinprors r#implicit_inputsOpRun.implicit_inputsFs%,,.tE{oQ?@ @%%D IIdii & 33K IIk&& '4;;C IIchh JJD[[ ! !ZZ>IIaL  E{r0c.URR$)zReturns node attribute `input`.)rrrIs r#r OpRun.input_s~~###r0c.URR$)z Returns node attribute `output`.)rr rIs r#r  OpRun.outputd~~$$$r0c.URR$)z!Returns node attribute `op_type`.)rrrIs r#r OpRun.op_typeis~~%%%r0c.URR$)z Returns node attribute `domain`.)rrQrIs r#rQ OpRun.domainnrr0cg)z Tells the runtime if this node needs the context (all the results produced so far) as it may silently access one of them (operator Scan, If, Loop). The default answer is `False`. Fr(rIs r# need_contextOpRun.need_contextssr0cURRS-SURR3/n[ UR R 55HFup#US;aM SUSs=::aS::dMO M#USS:wdM.URS US US 35 MH URS 5 S RU5$)NrFz op_type=>descrarzrz =,rGr) rHr*rrr__dict__rrjoin)rAattsrr"s r#__str__ OpRun.__str__|s''#-dnn>T>T=U/VW4==..01DA))ad!c!!aesl d1#QqcO, 2 Cyyr0cJ[SURR<S35e)a, Should be overwritten. :param args: operator inputs :param kwargs: optional inputs and overriden attributes, an attribute may be overridden if it belongs to a function, in this case, the same instance of OpRun can be called with different values of the same attribute. :return: outputs z/Method '_run' must be overwritten for operator rP)rRrHr*)rArkwargss r#_run OpRun._runs)"=dnn>U>U=XXY Z  r0)linked_attributesrc UR5(aUc[S[U5S35eOUb[S[U5S35eUR(aUc[ S[U5S35eUR(d'Ub$[ S[U5S[ U5S35e0nUR(aUc[ S [U5S35eURHen[X5n[U[5(dM%URU;a [ S U<S U<S [U5S35eXRXE'Mg URS URR[U55 0nURHOnXT;aM [!X5(d([#SU<SURR<S35e[X5Xu'MQ UR$(a4UR(aU(d[S[U5S35eXS'UbX'S'U(aUR&"U0UDUD6nOUR&"U0UD6nURSURR[U55 [U[,5(d0[)SURR<S[U5S35e[U5S:Xa$[ SURR<S35e[/SU55(aCUV s/sHn [U 5PM n n [)SURR<SU <S35eU$![([ 4aen [)SUV s/sHn [U 5PM Os sn fsn S[+U5S[+U5SURR<S3 5U eSn A ff=fs sn f) ac Calls method ``_run``, catches exceptions, displays a longer error message. :param args: inputs :param linked_attributes: used if this has an attriute linked to the attribute of the function it belongs to :param context: if this node is part of the subgraph, `context` is a dictionary with the values this node may use :return: tuple of results NzThis node if type z needs context to be filled.z+ does not need any contextbut one is given.z This node zK has linked attributes but None are given in parameter 'linked_attributes'.zM has no linked attribute but some are given in parameter 'linked_attributes' rPzBOne attribute is linked but no linked value is provided, in class z,Unable to find a value for linked attribute z in z in node z-- begin %s.run(%d inputs)rz is missing in operator z6A subgraph has linked attribute but none was given to rrzIssues with types z and attributes z and linked attributes=z (operator ).z-- done %s.run -> %d outputszMethod '_run' of class z does not return a tuple but rz does not return any result.c3B# UHn[U[5v M g7fr?)r[rw)rsrs r#ruOpRun.run..s1Sz!U##Ssz6One of the results returned by method '_run' of class zf is a tuple, this is no onnx correponding type (Map, List, Tensor, SparseTensor). All returned types: )rrr]rrrrrrr[r<rrrHr*lenr NameErrorrr-r\rrwany) rAr/rroverridden_attributesryr"r,rTerrdtypess r#r OpRun.runs     "(d 4PQ $T$ZL0[\   $ $):)BT$ZL(st ((->-JT$ZL)''*+<'=&>aA !#  $ $ ($ $T |1.--D&a--vv%66(J3'QUVgUjk''+Dzl!52C661J).. .0G0GTS))C+4%% '?@W@W?ZZ[\"$,FK*   ((1B"LTRVZLXYZ$5<   '9  $iiI)>I&Iii00 0$..2I2I3s8T#u%%)$..*A*A)DDabfgjbkallmn  s8q=)$..*A*A)DD`a  1S1 1 1'*+s!d1gsF+HI`I`Hcd''-j3   ->* $t%""9&AV:W9XY!^^447r;    ,s12M%M%$O%O5 O?N AOOcURnSU;a U[54$URSS5up[U5nX4$![a UR[54s$f=f)Nrr)r*r rrr)clsrr!i_verss r# infer_nameOpRun.infer_nameso|| d?+-- -[[a(  6YF| 6<` creates an onnx node returned by method :meth:`make_node `. .. exec_code:: import numpy as np from onnx.reference.ops._op_list import Celu onnx_node = Celu.make_node(alpha=0.5) print(onnx_node) Nxr)r> op_domainr min_input min_outputranger ) r<n_inputs n_outputsr,ropsetrQrUrnames_in names_outrs r#r OpRun.make_nodes6)  ~#GF;''H  ~#GF;))I%*8_5_asG_5&+I&67&6qW&6 7I@@ 67s *B(B-rc p^SU4SjjnUR"X40UD6nTUSS[50S.nU"Xg5nU$)a$ Instantiates this class based on the given information. :param n_inputs: number of inputs (default is defined by the operator schema) :param n_outputs: number of outputs (default is defined by the operator schema) :param verbose: verbosity :param kwargs: node attributes :return: NodeProto cB>TS:a[U[U5-5 gg)Nr)printrw)rrrs r# log_function"OpRun.create..log_function>s {gd +,r0Nr)rrrr)rrLrrrNro)r r ) r<rFrGrr,rOrrcls ` r#create OpRun.create,sJ$ -}}X;F;-/0   " r0rGrcUR"[U54XS.UD6nUR"U6n[U5S:XaUS$U$)z Evaluates this operator. :param args: inputs :param n_outputs: number of outputs (default is defined by the operator schema) :param verbose: verbosity :param kwargs: node attributes :return: NodeProto rTrr)rRr4r)r<rGrrr,instrTs r#eval OpRun.evalLsF"zz#d)TyTVThho s8q=q6M r0)rrrrrrr?)rrrzdict[str, Any]rUr)ryrrzAttributeProto | NonerNr)rNro)r rrNz list[str])rNz Iterable[str]rM)rNboolNN)rF int | NonerGr[r,rrNr)NNr) rFr[rGr[rrr,rrNr) rz list[Any]rGr[rrr,rrNr)/r*r+r,r-r.rBrFLOATFLOATSrGRAPHSINTINTS SPARSE_TENSORSPARSE_TENSORSSTRINGSTRINGSTENSORTENSORS TYPE_PROTO TYPE_PROTOSr__annotations__rBrr staticmethodrrrrprr rrQrr)abcabstractmethodr-r classmethodr>r rRrWr/r(r0r#rrsI ;N6I2C$$&U%%(  @ TC W!!#?""$V!A#%=(OS " 0> HK :/EI ! ,A  @EE+7Z0$$%%&&%%     ,0\|   $ $))) )  ))V $ $     >!%     r0rc,\rSrSrSrSSSjjrSrSrg) OpRunExpandidz0 Class any operator to avoid must inherit from. Nc2[S[U5S35eNz5The reference implementation must not use this node (r1rr])rArrOimpls r#rBOpRunExpand.__init__iCDJ t  !S\ 1A#g,P669$:K:K6L5MN LL+=qJ  W~r0)rrrZ)rrrOrrsrrzdict[str, Any] | None) r*r+r,r-r.rBr-rr/r(r0r#r|r|tsD,0      *  *=r0r|c,\rSrSrSrSSSjjrSrSrg) OpFunctionContextDependantiz The function can be instantiated but only at execution time. An instance of OpFunction is created everytime to node is executed. This is needed when the schema of an operator defines a context dependant function. Nc[RXX 0S9 X0lURURn[ UR XAR5Ulg)N)rsr)r|rBparentrrQr rschema_)rArrOrversions r#rB#OpFunctionContextDependant.__init__sKD\QST -- 0 01!)"3"3W>N>NO r0c/nUH=n[UR5nUR[XTR 55 M? UR"R%UR&U5nU"UR&UR(5nUR*"UR,/UQ70UD6$![anUR[:Xa[R nSnANUR[ :Xa[RnSnANUR[:Xa[RnSnAGNUR[:Xa[RnSnAGN;UR[:Xa[RnSnAGNeUeSnAff=fr?)r rlrrrrrrrrrrrBLOFAT16rr rir _load_implrrrr) rArxr,typesrttyper8rQrVs r#r-OpFunctionContextDependant._runsA 09 LL/ww? @!"[[ # #DNNE :$..$//2~~djj<6rs# **RRSS&II| , !4  ; ; &  00!!$L$NlCGGl^  %  ..b$=$=r0