wicddlmZddlZddlmZmZddlZddlZerddl m Z Gdde Z Gdde Z Gd d eZGd d ZGd dZd&dZeZGddZGddZd'dZd(dZGddZGddejZGd d!eZGd"d#eZGd$d%eZdS))) annotationsN) TYPE_CHECKINGAny)SequenceceZdZdZdS)RuntimeTypeErrorz/Raised when a type of a variable is unexpected.N__name__ __module__ __qualname____doc__e/root/.openclaw/workspace/chatterbox_venv_py311/lib/python3.11/site-packages/onnx/reference/op_run.pyrrs9999rrceZdZdZdS)RuntimeContextErrorzfRaised when the context is missing but an context dependent implementation is defined for an operator.Nr rrrrrspppprrceZdZdZdS)RuntimeImplementationErrorz8Raised when no implementation was found for an operator.Nr rrrrrsBBBBrrceZdZdZdS) DefaultNonezjDefault value for parameters when the parameter is not set but the operator has a default behavior for it.Nr rrrrrsttttrrc"eZdZdZddZd dZdS) RefAttrNamez}Implements a link between a parameter of a function and an attribute in node. Args: name: name of the input namestrc||_dSN)r)selfrs r__init__zRefAttrName.__init__(s  rreturnc0|jjd|jdS)N()) __class__r rrs r__repr__zRefAttrName.__repr__+s .)::DI::::rN)rrrr)r r r r rr%rrrrr!sF ;;;;;;rrrdict[str, onnx.defs.OpSchema]c i}tjjjD]}|j|vrt|j||jjkr3t d|jd|jd||jjd|j||jjkr |||j<n |||j<|||jdzt|jz<|S)Nz7This function assumes every operator has a unique name z even across multiple domains z and ._)onnxdefsget_all_schemas_with_historyrdomainNotImplementedError since_versionr)resschemas r_build_schemasr3/s)+C).%BBDD D D ;#  }FK 0 777)gfkgg4:MggJMfkJZJaggg#c&+&6&DDD#)FK %C =CFK# F$8 9 9 9:: JrceZdZddZd dZdS) OnnxType type_protoonnx.TypeProtoct|tjs tdt |d||_dS)Nz type_proto z must be of type onnx.TypeProto.) isinstancer+ TypeProto TypeErrortyper6)rr6s rrzOnnxType.__init__FsK*dn55 Pd:..PPP %rrrcd|jdS)Nz OnnxType(r")r6r$s rr%zOnnxType.__repr__Ms/4?////rN)r6r7r&)r r r rr%rrrr5r5Es<%%%%000000rr5c2eZdZdZd d Zedd Zd S) SparseTensorzcSimple representation of a sparse tensor. It is based on numpy but does not require scipy. values np.ndarrayindicesshape tuple[int]rNonec0||_||_||_dSr)r@rBrC)rr@rBrCs rrzSparseTensor.__init__Vs   rrc|jjSr)r@dtyper$s rrHzSparseTensor.dtype]s {  rN)r@rArBrArCrDrrE)rr)r r r r rpropertyrHrrrr?r?QsR!!!X!!!rr?attonnx.AttributeProtoctd|jD}ttj|jtj|j|S)zHosts a sparse tensor.c3K|]}|VdSrr).0ds r z#to_sparse_tensor..ds"&&!&&&&&&r)tupledimsr?r+ numpy_helperto_arrayr@rB)rJrCs rto_sparse_tensorrUbsa &&SX&&& & &E  ""3:.. ""3;//   r attr_type!onnx.AttributeProto.AttributeTypec tjjdtjjdtjjdtjjdtjjdtjjdtjjdtjj dtjj d tjj d tjj d tjj d tjjd tjjdi|S)Nc4tj|jSr)npfloat32frJs rz0_attribute_conversion_function..nsrz#%/@/@rc$d|jDS)Nc6g|]}tj|Sr)rZr[)rNr\s r zD_attribute_conversion_function....os 0S0S0S1A0S0S0Sr)floatsr]s rr^z0_attribute_conversion_function..os0S0S 0S0S0Src*t|jSr)Graphgr]s rr^z0_attribute_conversion_function..psuSU||rc$d|jDS)Nc,g|]}t|Sr)rd)rNres rrazD_attribute_conversion_function....qs0N0N0Naq0N0N0Nr)graphsr]s rr^z0_attribute_conversion_function..qs0N0N3:0N0N0Nrc*t|jSr)intir]s rr^z0_attribute_conversion_function..rsSZZrc$d|jDS)Nc,g|]}t|Sr)rjrNrks rrazD_attribute_conversion_function....ss.H.H.H!s1vv.H.H.Hr)intsr]s rr^z0_attribute_conversion_function..ss.H.Hsx.H.H.Hrc*t|jSr)rU sparse_tensorr]s rr^z0_attribute_conversion_function..ts7G  8 8 rc$d|jDS)Nc,g|]}t|Sr)rUrNts rrazD_attribute_conversion_function....ws.9 9 9 $% Q  9 9 9 r)sparse_tensorsr]s rr^z0_attribute_conversion_function..ws#9 9 ),);9 9 9 rc6|jdS)Nutf-8)sdecoder]s rr^z0_attribute_conversion_function..zs W0E0Erc$d|jDS)Nc8g|]}|dS)rx)rz)rNrys rrazD_attribute_conversion_function....{s12 2 2 "#AHHW  2 2 2 r)stringsr]s rr^z0_attribute_conversion_function..{s"2 2 '*{2 2 2 rcJtj|jSr)r+rSrTrur]s rr^z0_attribute_conversion_function..~s0A0J0J350Q0Qrc$d|jDS)NcLg|]!}tj|"Sr)r+rSrTrts rrazD_attribute_conversion_function....s62 2 2 ./D  & &q ) )2 2 2 r)tensorsr]s rr^z0_attribute_conversion_function..s"2 2 36;2 2 2 rc*t|jSr)r5tpr]s rr^z0_attribute_conversion_function..sHSV4D4Drc$d|jDS)Nc,g|]}t|Sr)r5rts rrazD_attribute_conversion_function....s+6 6 6 HQKK6 6 6 r) type_protosr]s rr^z0_attribute_conversion_function..s"6 6 !$6 6 6 r)r+AttributeProtoFLOATFLOATSGRAPHGRAPHSINTINTS SPARSE_TENSORSPARSE_TENSORSSTRINGSTRINGSTENSORTENSORS TYPE_PROTO TYPE_PROTOS)rVs r_attribute_conversion_functionrls !#@#@ "$S$S !#;#; "$N$N !7!7  "H"H ), ,  *- -  "$E$E #& &  "$Q$Q #& &  &(D(D '* * - 23 rceZdZdZddZdS) rdrereonnx.GraphProtorrEc||_dSrr)rres rrzGraph.__init__s rN)rerrrE)r r r __slots__rrrrrdrds-IrrdceZdZdZdZ d6d7d Zd Z d6d8dZedZ d9dZ ed:dZ e d;dZ e d;dZe dd%Zddd&d'Zed(Ze d?d@d-Ze dAdBd1Zedd.d2dCd5ZdS)DOpRuna0Ancestor to all operators in this subfolder. Args: onnx_node: `onnx` node 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 schema: operator schema N onnx_nodeonnx.NodeProto run_paramsdict[str, Any]r2rc \t|ts tdt|ddD])}||vr#t d|dt |d*d|vrt d||_||_|t|j d r|j j |_ ne|j j tvrt|j j |_ n5|jtvrt|j|_ nd|_ n||_ d |_|dS) Nz$run_params must be a dictionary not r))opsetsnew_ops Attribute z must be in run_params, only z was found.logz#run_params must contains key 'log'. op_schemaF)r9dictr;r< RuntimeErrorsortedKeyErrorrrhasattrr#r_schemar _schemasop_type has_subgraph_load_attributes)rrrr2rJs rrzOpRun.__init__sX*d++ XV4 CSCSVVVWW W(  C*$$"777j))777%  " "@AA A"$ >t~{33 $#~7 (H44'(?@ "h..' (9: # !DL! rc.|jd|g|RdS)Nr)r)rpatternargss r_logz OpRun._logs'w.......rrJrKref_attonnx.AttributeProto | NonercJ|jtjjkr|jdd}d|jvr-t |jd}nd}|jdd}| Jd|||j|jdtd|jddd z |dn t || St|j}| ||S|=td |j d |jd |j jd|j jd|d td|j d |jd |j jd|j jd|d|d )z0Converts an attribute value into a python value.rNexisting_functions evaluator_clsz0evaluator_cls must be specified to evaluate att=rrverbose)rrr functionszUnable to convert attribute z type z from node type z , domain  r)z$Unable to convert default value for )r<r+rrrgetlistr@remaxrAttributeErrorrrrr.)rrJrrrrconversion_functions r_extract_attribute_valuezOpRun._extract_attribute_values$ 8t*0 0 0o)))T::G#t66 1E!F!M!M!O!OPP   O//FFM ,,H3HH-,,!=x0At229a@@1DEE 'T'..:J:J5K5K#  =SXFF  *&&s++ + ? .s!%*%=%= &&r>Constantrequiredrz3 is expected based on ONNX specifications for node r) default_valuer)has_linked_attributer attributer ref_attr_namerrsetattrappendr<r+rrrrrritemsrgetattrrrru data_typesetattributes_names_)radded_attributesrJrrkvs rrzOpRun._load_attributess?$)!>+  C8D  ;#C$566,0))55c:: D$ & & &  # #D ) ) )x4.444$(!))U-GG))&CH&&# < 3DN2,FF /5577 3 31tQ''3q*d33*DDD(,(>DDDq/22 3?/144O0A55 ! 1 ;q @ @%)EE$($A$A!/ST$U$UEa///(//222!$%5!6!6rgraphr list[str]c<t|tjs tdt |dt }t }|jD]}||j|j D]}||j|j D]}||j|j D]D}|j D]}|||j D]}||vr||Et|S)zReturns all variables 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 r))r9r+ GraphProtor;r<r initializeraddrsparse_initializerinputnodeoutputr) rlocalknowninit sparse_initinprorks rimplicit_inputszOpRun.implicit_inputss6 %11 A?tE{{???@@ @% ! !D IIdi  3 ( (K IIk& ' ' ' ';  C IIch    J ! !D[   ! Z ! !E>>IIaLLL !E{{r Sequence[str]c|jjS)zReturns node attribute `input`.)rrr$s rrz OpRun.input/s~##rc|jjS)z Returns node attribute `output`.)rrr$s rrz OpRun.output4~$$rrc|jjS)z!Returns node attribute `op_type`.)rrr$s rrz OpRun.op_type9s~%%rc|jjS)z Returns node attribute `domain`.)rr.r$s rr.z OpRun.domain>rrboolcdS)zTells 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`. Frr$s r need_contextzOpRun.need_contextCs urcp|jjdzd|jjg}t |jD]H\}}|dvr d|dcxkrdkr+n |ddkr|d |d |d I|d d |S)Nr!z op_type=>descrarzr*z =,r"r) r#r rrr__dict__rrjoin)rattsrrs r__str__z OpRun.__str__Ks'#-/Vdn>T/V/VW4=..0011 - -DAq)))ad!!!!c!!!!!aesll O1OOqOOO,,, Cyyrc<td|jjd)aShould be overwritten. Args: *args: operator inputs **kwargs: optional inputs and overridden 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. Returns: outputs z/Method '_run' must be overwritten for operator r))r/r#r rrkwargss r_runz OpRun._runUs)" Zdn>U Z Z Z   rr1tuple[Any, ...]ct|ts-td|jjdt |d|st d|jjdtd|Dr,d|D}td|jjd|d td |D}td |Dr,d |D}td|jjd |d |S)z-Checks the output are from the expected type.zMethod '_run' of class z does not return a tuple but 'z'.z does not return any result.c3@K|]}t|tVdSr)r9rQrts rrPz/OpRun._check_and_fix_outputs..qs,11z!U##111111rc,g|]}t|Srr<rts rraz0OpRun._check_and_fix_outputs..r+++!d1gg+++rz6One of the results returned by method '_run' of class zg is a tuple, this is no ONNX corresponding type (Map, List, Tensor, SparseTensor). All returned types: r)c3jK|].}tj|rtj|n|V/dSr)rZisscalararray)rNxs rrPz/OpRun._check_and_fix_outputs..xsL  78BKNN 1RXa[[[      rc3K|]<}t|tjttfpt |d V=dS)todenseN)r9rZndarrayrrrrts rrPz/OpRun._check_and_fix_outputs..{s[  A D$788QGAy.rrzs has an unexpected type, this is no ONNX corresponding type (Map, List, Tensor, SparseTensor). All returned types: )r9rQr;r#r r< ValueErrorany)rr1dtypess r_check_and_fix_outputszOpRun._check_and_fix_outputsgs#u%% p$.*Appcghkclclppp  a$.*Aaaa  11S111 1 1 ++s+++F3I`33'-333    .s%<%<% %d outputsr)rrr<rrrrrrr9rrrr#r lenr NameErrorrr r;rr) rrrroverridden_attributesrJrr r1es rrz OpRun.runs      "QdQQQ \T$ZZ\\\   $ ):)BtT$ZZttt ( ->-JAT$ZZAA'*+<'='=AAA !#  $ K ($. $T ...- K KD#&&a--Kv%666(5355Vg55'+Dzz5552C161J)#. .0GTSSS) - -C+++4%% \\\@W\\\"$,,F3KK   5( 1B "ZTRVZZZZZ$5F<   'F9  $ 1diI)>I&IIdi000>*   ;%<%>;;:@AV:W:W;;!^4;;;    * N #CHHHQ   **3///sH44JA JJc|j}d|vr |tjfS|dd\}} t |}n5#t $r(|jtjfcYSwxYw||fS)Nr*)r r+r,onnx_opset_versionrsplitrjr)clsrversi_verss r infer_namezOpRun.infer_names| d??55777 7[[a(( d @YYFF @ @ @<!=!=!?!?? ? ? ? @V|sA/BBn_inputs int | None n_outputsr c |\}}|j}d}|*|!tj|||}|j}|*|!tj|||}|j}dt|D}dt|D} tjj ||| fi|S)aCreates an ONNX node for this class based on the given information. Args: n_inputs: number of inputs (default is defined by the operator schema) n_outputs: number of outputs (default is defined by the operator schema) verbose: verbosity **kwargs: node attributes Returns: NodeProto Method :meth:`eval ` 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) Ncg|]}d|S)rrrns rraz#OpRun.make_node..s555GGG555rcg|]}d|S)yrrns rraz#OpRun.make_node..s777WWW777r) r, op_domainr+r, get_schema min_input min_outputrangehelper make_node) r)r-r/r ropsetr.r2names_in names_outs rr:zOpRun.make_nodes>))  ~--gufEE'H  ~--gufEE)I55U8__55577eI&6&6777 {$Wh LLVLLLrrrrjc d fd }|j||fi|}|dd tjid }|||S) aInstantiates this class based on the given information. Args: n_inputs: number of inputs (default is defined by the operator schema) n_outputs: number of outputs (default is defined by the operator schema) verbose: verbosity **kwargs: node attributes Returns: NodeProto rrrrrrEcVdkr!t|t|zdSdS)Nr&)printrQ)rrrs r log_functionz"OpRun.create..log_function-s2{{gd +,,,,,{rNr)rrrr)rrrrrrE)r:r+r,r')r)r-r/rr rArrs ` rcreatez OpRun.creates}, - - - - - -s}Xy;;F;;497799:   s4$$$rr/rr list[Any]c|jt|f||d|}|j|}t|dkr|dS|S)aEvaluates this operator. Args: *args: inputs n_outputs: number of outputs (default is defined by the operator schema) verbose: verbosity **kwargs: node attributes Returns: NodeProto rCr&r)rBr!r)r)r/rrr instr1s revalz OpRun.eval:sT(sz#d))Ty'TTVTTdho s88q==q6M rr)rrrrr2r)rJrKrrrr)rrE)rrrr)rrr&)rr)r1r rr NN)r-r.r/r.r rrr)NNr) r-r.r/r.rrjr rrr) rrDr/r.rrjr rrr)r r r r r4rrr staticmethodrrrrIrrrr.rrabcabstractmethodr rr classmethodr,r:rBrGrrrrrs{ITX     :///OS% % % % % NEE\E-7-7-7-7^\.$$$X$%%%X%&&&X&%%%X%    "@,0S0S0S0S0S0j  [  $ $,M,M,M,M[,M\ $ $ % % % %[ %D!% [rrceZdZdZdZdZdS) OpRunExpandz.Class any operator to avoid must inherit from.cBtdt|dNz5The reference implementation must not use this node (r rr<rs rrzOpRunExpand.__init__X( RDJJ R R R   rcBtdt|drPrQrinputsr s rr zOpRunExpand._run]rRrNr r r r rr rrrrNrNUs888        rrNc,eZdZdZ ddd Zd ZdZdS) OpFunctionzRuns a custom function.Nrrrdict[str, Any] | NoneimplrrrrEc| td|jd|jdt|||_fdt jd|D_dS)Nz"impl cannot be None for node type from domain r)c2i|]}|t|Sr)r)rNrrs r z'OpFunction.__init__..vs5    '$%%   r attributes_)rrr.rrimpl_rr_)rrrrZrs` rrzOpFunction.__init__fs <5Y5F55(/555  tY 333      M:FF   rc.|j|jg|Ri|Sr) _run_implr`rTs rr zOpFunction._run{s&t~dj<6<<> t ! !S\\ 1 1J#g,,JJ69$:K6L6LJJ LJJ9=JJJ  W~~rrH) rrrrYrZrrrYrrE)r r r r rr rbrrrrXrXcsY!! ,0      *===rrXc$eZdZdZ d d d Zd ZdS) OpFunctionContextDependantzThe 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 dependent function. NrrrrYparentrct||||i||_|j|j}t j|j||j|_ dS)N)rZr) rXrrnrr.r+r,r5rschema_)rrrrnversions rrz#OpFunctionContextDependant.__init__sc D)ZdrRRR - 01y++  w (8   rc`g}|D]^}tj|j}|tj||j_|j|j |}||j |j }|j |j g|Ri|Sr) r+r9np_dtype_to_tensor_dtyperHrmake_tensor_type_protorCrn _load_implrrrbr`)rrUr typesrurHclrFs rr zOpFunctionContextDependant._runs M MAK88AAE LL;;E17KK L L L L [ # #DNE : :r$.$/22t~dj<6<<r}s#""""" %%%%%%%% )((((((:::::|:::qqqqq,qqqCCCCC!4CCCuuuuuuuu ; ; ; ; ; ; ; ;& >   0 0 0 0 0 0 0 0!!!!!!!!":BBBBBCGBBBJ      %    ,,,,,,,,^==========r