o åetã @sÐddlmZddlmZmZmZmZmZmZm Z ddl Z ddl Z ddl mmZddl mZmZmZGdd„dƒZGdd„dƒZd ed eed ed ed eeedff dd„ZGdd„dƒZGdd„dƒZdS)é)Ú namedtuple)ÚSequenceÚAnyÚTypeÚTupleÚNewTypeÚOptionalÚDictN)Ú ModelProtoÚ NodeProtoÚ IR_VERSIONc@s8eZdZUedeƒZedƒZeed<edƒZeed<dS)Ú DeviceTypeÚ_TyperÚCPUéÚCUDAN) Ú__name__Ú __module__Ú __qualname__rÚintrrÚ__annotations__r©rrúAD:\Projects\ConvertPro\env\Lib\site-packages\onnx/backend/base.pyr s r c@s"eZdZdZdeddfdd„ZdS)ÚDevicez~ Describes device type and device id syntax: device_type:device_id(optional) example: 'CPU', 'CUDA', 'CUDA:1' ÚdeviceÚreturnNcCsB| d¡}tt|dƒ|_d|_t|ƒdkrt|dƒ|_dSdS)Nú:rr)ÚsplitÚgetattrr ÚtypeZ device_idÚlenr)ÚselfrÚoptionsrrrÚ__init__s  ÿzDevice.__init__)rrrÚ__doc__Ústrr#rrrrrsrÚtypenameÚ field_namesÚargsÚkwargsr.cs`dd„t|ƒDƒ‰| dd¡t||g|¢Ri|¤Ž}dtdtdtf‡fdd „ }t|d |ƒ|S) NcSsi|]\}}||“qSrr)Ú.0ÚiÚnrrrÚ #sz"namedtupledict..ÚrenameTr!Úkeyrcs&t|tƒr ˆ|}tt|ƒ|ƒ |¡S©N)Ú isinstancer%ÚsuperrÚ __getitem__)r!r/©Zfield_names_maprrÚgetitem(s znamedtupledict..getitemr3)Ú enumerateÚ setdefaultrrÚsetattr)r&r'r(r)Údatar5rr4rÚnamedtupledict"s   r:c@s*eZdZdededeedffdd„ZdS)Ú BackendRepÚinputsr)r.cKsdSr0r)r!r<r)rrrÚrun1szBackendRep.runN)rrrrrr=rrrrr;0s"r;c@sòeZdZe ddedededefdd„ƒZe ddededede e fdd „ƒZ e dded ededede ed ff d d „ƒZ e  dded edede ee eje ed ffdeeefde e ed ff dd„ƒZededefdd„ƒZdS)ÚBackendrÚmodelrr)rcKódS)NTr©Úclsr?rr)rrrÚ is_compatible6szBackend.is_compatiblecKstj |¡dSr0)ÚonnxÚcheckerZ check_modelrArrrÚprepare?s zBackend.preparer<.cKs*|j||fi|¤Ž}|dusJ‚| |¡Sr0)rFr=)rBr?r<rr)ÚbackendrrrÚ run_modelIs  zBackend.run_modelNÚnodeÚ outputs_infocKsFd|vrt ¡}t|_d|di|_tj ||¡dStj |¡dS)aˆSimple run one operator and return the results. Args: outputs_info: a list of tuples, which contains the element type and shape of each output. First element of the tuple is the dtype, and the second element is the shape. More use case can be found in https://github.com/onnx/onnx/blob/main/onnx/backend/test/runner/__init__.py Z opset_versionÚN)Ú c_checkerZCheckerContextr Z ir_versionZ opset_importsrDrEZ check_node)rBrIr<rrJr)Zspecial_contextrrrÚrun_nodeTs ÿzBackend.run_nodecCr@)zŽ Checks whether the backend is compiled with particular device support. In particular it's used in the testing suite. Tr)rBrrrrÚsupports_devicemszBackend.supports_device)r)rN)rrrÚ classmethodr r%rÚboolrCrr;rFrrHr rÚnumpyZdtyperr rMrNrrrrr>5slþÿþý üþÿþý ü ýÿþýü  û üÿþýü û úr>)Ú collectionsrÚtypingrrrrrrr rQZ onnx.checkerrDZonnx.onnx_cpp2py_export.checkerZonnx_cpp2py_exportrErLr r r r rr%r:r;r>rrrrÚs $.