2jkddlmZddlZddlmZmZddlZddlZerddl 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__FC:\Projects\Historical\venv\Lib\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 d6d8dZed6dZ d9dZ ed:dZ e d;dZ e d;dZe dd%Zdddd&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++ + ? .s00%XVVrConstantrequiredrz3 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&&*/D4 < 3DN2j@@ /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.input5s~##rc|jjS)z Returns node attribute `output`.)rrr$s rrz OpRun.output:~$$rrc|jjS)z!Returns node attribute `op_type`.)rrr$s rrz OpRun.op_type?s~%%rc|jjS)z Returns node attribute `domain`.)rr.r$s rr.z OpRun.domainDrrboolcdS)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_contextIs 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__Qs'#-/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._run[s)" 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..ws,11z!U##111111rc,g|]}t|Srr<rts rraz0OpRun._check_and_fix_outputs..x+++!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..~s;HHqBKNN9RXa[[[HHHHHHrc3K|]<}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_outputsms#u%% p$.*Appcghkclclppp  a$.*Aaaa  11S111 1 1 ++s+++F3I`33'-333  HHCHHHHH        ,+s+++F3I`33'-333   r)linked_attributesrrc |r#| tdt|dn"| tdt|d|jr"| t dt|d|js2|0t dt|dt |di}|jr| t d t|d|jD]f}t||}t|tr?|j |vr&t d |d |d t|d||j ||<g| d |j jt|i}|jD]J}||vrt!||s t#d|d|j jdt||||<K|jr.|jr"|s tdt|d||d<|||d<|!|jst|t&r||d< |r|j|i||} n |j|i|} nd#t*t f$rP} t+dd|Ddt-|dt-|d|j jd | d} ~ wwxYw| d|j j| t| nd|| S)aICalls method ``_run``, catches exceptions, displays a longer error message. Args: *args: inputs linked_attributes: used if this has an attribute linked to the attribute of the function it belongs to context: if this node is part of the subgraph, `context` is a dictionary with the values this node may use bindings: if not None, the active `SymbolBindings` instance (see `onnx.reference.shape_annotation_checker`) used to check shape annotations; propagated to any subgraph this node may invoke (If, Loop, Scan) so a symbolic dimension shared between the parent graph and the subgraph is checked consistently Returns: 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' r)zBOne 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 rrrzIssues with types c,g|]}t|Srr)rNr*s rrazOpRun.run..s%<%<% %d outputsr)rrr<rrrrrrr9rrrr#r lenr NameErrorr OpFunctionr r;rr) rrrrroverridden_attributesrJrr r1es rrz OpRun.runsF(      "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   T%6 *T::V:V !)F:  $ 1diI)>I&IIdi000>*   ;%<%>;;:@AV:W:W;;!^4;;;    * N #CHHHQ   **3///s7IJ8(A J33J8c|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/r1r 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_function9s2{{gd +,,,,,{rNr)rrrr)rrrrrrE)r<r+r,r))r+r/r1rr rCrrs ` rcreatez OpRun.create#s}, - - - - - -s}Xy;;F;;497799:   s4$$$rr1rr 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 rEr(r)rDr"r)r+r1rrr instr1s revalz OpRun.evalFsT(sz#d))Ty'TTVTTdho s88q==q6M rr)rrrrr2r)rJrKrrrr)rrE)rrrr)rrr&)rr)r1r rr NN)r/r0r1r0r rrr)NNr) r/r0r1r0rrjr rrr) rrFr1r0rrjr rrr)r r r r r6rrr staticmethodrrrrIrrrr.rrabcabstractmethodr rr classmethodr.r<rDrIrrrrrsITX     :///OS% % % % % N S S S\ S+7+7+7+7Z\.$$$X$%%%X%&&&X&%%%X%    "<,0[0[0[0[0[0z  [  $ $,M,M,M,M[,M\ $ $ % % % %[ %D!% [rrceZdZdZdZdZdS) OpRunExpandz.Class any operator to avoid must inherit from.c(td|dNz5The reference implementation must not use this node (r!)r)r+rr s r__new__zOpRunExpand.__new__ds" KC K K K   rcBtdt|drR)rr<rinputsr s rr zOpRunExpand._runis( RDJJ R R R   rN)r r r r rSr rrrrPrPas888        rrPc,eZdZdZ ddd Zd ZdZdS)r$zRuns 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__..s5    '$%%   r attributes_)rrr.rrimpl_rr^)rrrrYrs` rrzOpFunction.__init__rs <5Y5F55(/555  tY 333      M:FF   rc.|j|jg|Ri|Sr) _run_implr_rUs rr zOpFunction._runs&t~dj<6<<r}s#""""" %%%%%%%% )((((((:::::|:::qqqqq,qqqCCCCC!4CCCuuuuuuuu ; ; ; ; ; ; ; ;& >   0 0 0 0 0 0 0 0!!!!!!!!":NNNNNCGNNNb      %    44444444n==========r