o ˜ÄŽi< ã@sfddlmZddlmZddlmZddlmZddlmZddlmZddl m Z Gdd „d ƒZ d S) é)ÚMappingProxyType)ÚAny)ÚMapping)ÚTuple)ÚAttributeProto)Ú NodeProto©Ú OnnxTensorc@sÜeZdZdedefdd„Zededefdd„ƒZ e defd d „ƒZ e defd d „ƒZ e defd d„ƒZ e defdd„ƒZe defdd„ƒZe deedffdd„ƒZe deedffdd„ƒZe deeeffdd„ƒZdS)ÚOnnxNodeÚonnx_node_protoÚ unique_namecCsB||_||_t|jƒ|_t|jƒ|_d|_dd„|jjDƒ|_ dS)NcSsi|] }|jt |¡“qS©)Únamer Ú_parse_attribute_value)Ú.0Ú attributer r úP/home/jeff/fluffinator/venv/lib/python3.10/site-packages/onnx2torch/onnx_node.pyÚ sÿz%OnnxNode.__init__..) Ú_protoÚ _unique_nameÚtupleÚinputÚ _input_valuesÚoutputÚ_output_valuesÚ_inputsrÚ_proto_attributes)Úselfr r r r rÚ__init__ s   ÿzOnnxNode.__init__rÚreturncCs¾| d¡r |j}|S| d¡r|j}|S| d¡r!t|jdƒ}|S| d¡r-t|jƒ}|S|jr7t|jƒ}|S|j rAt|j ƒ}|S|j rNdd„|j Dƒ}|S|j r[dd„|j Dƒ}|S|}|S) NÚiÚfÚsúutf-8ÚtcSsg|]}t|dƒ‘qS)r#)Ústr)rr"r r rÚ 'sz3OnnxNode._parse_attribute_value..cSsg|]}t|ƒ‘qSr r)rr$r r rr&)s) ÚHasFieldr r!r%r"r r$ÚintsÚlistÚfloatsÚstringsÚtensors)rÚvaluer r rrs4  ï ñ  ó  õ  ÷ ùûþzOnnxNode._parse_attribute_valuecCó|jS©N)r©rr r rÚproto/ózOnnxNode.protocCó|jjSr/)rrr0r r rr3óz OnnxNode.namecCr.r/)rr0r r rr 7r2zOnnxNode.unique_namecCr3r/)rÚdomainr0r r rr5;r4zOnnxNode.domaincCr3r/)rÚop_typer0r r rÚoperation_type?r4zOnnxNode.operation_type.cCr.r/)rr0r r rÚ input_valuesCr2zOnnxNode.input_valuescCr.r/)rr0r r rÚ output_valuesGr2zOnnxNode.output_valuescCs t|jƒSr/)rrr0r r rÚ attributesKs zOnnxNode.attributesN)Ú__name__Ú __module__Ú __qualname__rr%rÚ staticmethodrrrÚpropertyr1rr r5r7rr8r9rr:r r r rr s( r N) ÚtypesrÚtypingrrrÚonnx.onnx_ml_pb2rrÚonnx2torch.onnx_tensorr r r r r rÚs