o Xi8@sdZddlmZddgZddlZddlZddlmZmZddl m Z m Z ddl Z ddl m Z mZe d ZeeZGd d d ejd ZGd ddeZGdddeZGdddejed fZGdddejedfZdS)zTracked containers for graph.) annotations GraphInputs GraphOutputsN)IterableSequence) SupportsIndexTypeVar)_core _protocolsTcseZdZdZd*d+fdd Zd,d d Zd-d dZd-ddZd.fdd Zd,fdd Z d/fdd Z d0d1fdd Z d.fdd Z d,fdd Z d2d"d#Zd,fd$d% Zd&d'Zd(d)ZeZeZeZeZeZZS)3_GraphIOz"The inputs and outputs of a Graph.Ngraph _core.GraphcsL||_t|_|durt|}|D]}||qt||dSN) _graph collectionsCounter _ref_countertuple _set_graphsuper__init___check_invariance)selfr initlistvalue __class__M/home/ubuntu/.local/lib/python3.10/site-packages/onnx_ir/_graph_containers.pyrs    z_GraphIO.__init__returnNonecCt)"Check the invariance of the graph.NotImplementedErrorrrrrr.z_GraphIO._check_invariancer _core.ValuecCr")Set the graph for the value.r$rrrrrr2r'z_GraphIO._set_graphcCr")Unset the graph for the value.r$r*rrr_maybe_unset_graph6r'z_GraphIO._maybe_unset_graphitemcs"||t||dS)zAdd a new input to the graph.N)rrappendrrr-rrrr.:s   z_GraphIO.appendcs,t|}|D]}||qt|dS)z%Extend the list of inputs or outputs.N)rrrextend)rotherr-rrrr0As z_GraphIO.extendiintcs$t|||||dS)z$Insert an input/output to the graph.N)rinsertrr)rr2r-rrrr4Hs  z_GraphIO.insertcs"t|}||||S)&Remove an input/output from the graph.)rpopr,r)rr2rrrrr7Ns  z _GraphIO.popcs"t||||dS)r6N)rremover,rr/rrrr8Us   z_GraphIO.removecs$|jD]}||qtdS)zClear the list.N)datar,rclearr*rrrr:[s  z_GraphIO.clearlist[_core.Value]cCs |jS)z"Return a shallow copy of the list.)r9copyr&rrrr<as z _GraphIO.copycst|tr.t|tr.|j|D]}||q|D]}||qt|||dSt|t rM||j|||t|||dSt dt |dt |)z$Replace an input/output to the node.NzInvalid types for __setitem__: z and ) isinstancerslicer9r,rr __setitem__rr TypeErrortype)rr2r-rrrrr?fs    z_GraphIO.__setitem__cCs |j|S)z#Get an input/output from the graph.)r9)rr2rrr __getitem__{s z_GraphIO.__getitem__cOstd)zUnimplemented method.zMethod is not supported) RuntimeError)r_args_kwargsrrr_unimplementedsz_GraphIO._unimplementedrr rr r!rr(r r!)r-r(r r!)r2r3r-r(r r!)r5)r2r3r r()r r;)__name__ __module__ __qualname____doc__rrrr,r.r0r4r7r8r:r<r?rBrF__add____radd____iadd____mul____rmul__ __classcell__rrrrr s*     r r(c@.eZdZdZd ddZddd Zdd d Zd S)rzThe inputs of a Graph.r r!cC<tjsdS|jD]}|j|jurqtd|d|jdS)r#NInvariance error: Value 'z ' is not an input of the graph: onnx_irDEBUGr9r ValueErrorr*rrrr  zGraphInputs._check_invariancerr(cCsf|jdur|j|jurtd|d|dur!td|d|j|d7<d|_|j|_dS)r)NValue 'z^' is already owned by a different graph. Please remove the value from the previous graph firstzf' is produced by a node and cannot be an input to the graph. Please create new Values for graph inputsT)rrZproducerr_is_graph_inputr*rrrrs    zGraphInputs._set_graphcCT|j|jus Jd|j|d8<|j|dkrdSd|_|r%dSd|_dSr+'Bug: value does not belong to the graphr]rNF)rrr__owned_by_graphr*rrrr, zGraphInputs._maybe_unset_graphNrHrIrJrKrLrMrrr,rrrrrs   c@rT)rzThe outputs of a Graph.r r!cCrU)r#NrVz!' is not an output of the graph: rWr*rrrrr[zGraphOutputs._check_invariancerr(cCsJ|jdur|j|jurtd|d|j|d7<d|_|j|_dS)r)Nr\zb' is already an output of a different graph. Please remove the value from the previous graph firstr]T)rrZr_is_graph_outputr*rrrrs  zGraphOutputs._set_graphcCr`ra)rrrfrcr*rrrr,rdzGraphOutputs._maybe_unset_graphNrHrIrerrrrrs    cs^eZdZdZddfdd Zdd d Zdd dZdfdd Zdfdd ZdddZ Z S)GraphInitializerszUThe initializers of a Graph as ``dict[str, Value]`` with additional mutation methods.Nr rc sRi}|dur |||r||||_|D]}||qt|dSr)updatervaluesrrr)rr dictkwargsr9rrrrrs    zGraphInitializers.__init__rr(r r!cCs8|jdur|j|jurtd|dd|_|j|_dS)r)Nr\zg' is already an initializer of a different graph. Please remove the value from the previous graph firstT)rrZ_is_initializerr*rrrrs   zGraphInitializers._set_graphcCs0|j|jus Jdd|_|rdSd|_dS)r+rbFN)rrlrcr*rrrr,s  z$GraphInitializers._maybe_unset_graphkeystrcst|tjstdt|t|tstdt||dkr%td|js3t d||||_n||jkrDtd|d|jd| d urRtd |d ||j vra|j |}| || |t||d S) z!Set an initializer for the graph.z"value must be a Value object, not z!Value name must be a string, not z$Value name cannot be an empty stringz1Value %s does not have a name, setting it to '%s'zKey 'z(' does not match the name of the value 'z('. Please use the value.name as the key.Nr\z9' is produced by a node and cannot be a graph initializer)r=r Valuer@rArnrZnameloggerinfor^r9r,rrr?)rrmr old_valuerrrr?s,         zGraphInitializers.__setitem__cs$|j|}||t|dS)z%Delete an initializer from the graph.N)r9r,r __delitem__rrmrrrrrus  zGraphInitializers.__delitem__cC|||j<dS)z Add an initializer to the graph.Nrqr*rrradd zGraphInitializers.addrrGrI)rmrnrr(r r!)rmrnr r!) rJrKrLrMrrr,r?ruryrSrrrrrgs  rgcseZdZdZd3fdd Zd4fdd Zd5ddZd6d7ddZd6d8ddZd6d9ddZ d6d:dd Z d6d;d"d#Z d6dd+d,Z d6d?d.d/Zd6d@d1d2ZZS)A AttributeszOThe attributes of a Node as ``dict[str, Attr]`` with additional access methods.attrsIterable[_core.Attr]owner_core.Node | _core.Functionr r!cs ||_tdd|DdS)NcSsi|]}|j|qSrrx).0attrrrr ,sz'Attributes.__init__..)_ownerrr)rr|r~rrrr(szAttributes.__init__rmrnr _core.AttrcsNt|turtdt|t|tjstdt|t||dS)zSet an attribute for the node.zKey must be a string, not zValue must be an Attr, not N)rArnr@r=r Attrrr?rvrrrr?.s  zAttributes.__setitem__cCrw)zAdd an attribute to the node.Nrxr*rrrry6rzzAttributes.addNdefaultr int | TcC||vr ||S|S)z'Get the integer value of the attribute.)as_intrrmrrrrget_int: zAttributes.get_int float | TcCr)z%Get the float value of the attribute.)as_floatrrrr get_float@rzAttributes.get_floatstr | TcCr)z&Get the string value of the attribute.) as_stringrrrr get_stringFrzAttributes.get_string_protocols.TensorProtocol | TcCr)z&Get the tensor value of the attribute.) as_tensorrrrr get_tensorLrzAttributes.get_tensor_core.Graph | TcCr)z%Get the graph value of the attribute.)as_graphrrrr get_graphRrzAttributes.get_graphSequence[int] | TcCr)z0Get the Sequence of integers from the attribute.)as_intsrrrrget_intsXrzAttributes.get_intsSequence[float] | TcCr)z.Get the Sequence of floats from the attribute.) as_floatsrrrr get_floats^rzAttributes.get_floatsSequence[str] | TcCr)z/Get the Sequence of strings from the attribute.) as_stringsrrrr get_stringsdrzAttributes.get_strings'Sequence[_protocols.TensorProtocol] | TcCr)z/Get the Sequence of tensors from the attribute.) as_tensorsrrrr get_tensorsjs zAttributes.get_tensorsSequence[_core.Graph] | TcCr)z.Get the Sequence of graphs from the attribute.) as_graphsrrrr get_graphstrzAttributes.get_graphs)r|r}r~rr r!)rmrnrrr r!)rrr r!r)rmrnrr r r)rmrnrr r r)rmrnrr r r)rmrnrr r r)rmrnrr r r)rmrnrr r r)rmrnrr r r)rmrnrr r r)rmrnrr r r)rmrnrr r r)rJrKrLrMrr?ryrrrrrrrrrrrSrrrrr{%s            r{r)rM __future__r__all__rloggingcollections.abcrrtypingrrrXr r r getLoggerrJrrUserListr rrUserDictrnrgr{rrrrs$  o*&K