a ==ic ã@s<ddlmZddlmZddlmZmZGdd„deƒZdS)é)Úcore)Ú caffe2_pb2)Ú BackendRepÚnamedtupledictcs4eZdZ‡fdd„Zedd„ƒZ‡fdd„Z‡ZS)Ú Caffe2Repcs6tt|ƒ ¡||_||_||_||_d|_d|_dS)NF) ÚsuperrÚ__init__Úinit_netÚ predict_netÚ workspaceÚ uninitializedÚ nets_createdÚ ran_init_net)Úselfr r r r ©Ú __class__©úo/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/caffe2/python/onnx/backend_rep.pyr szCaffe2Rep.__init__cCs$|jjjtjkr d |jjj¡SdS)Nzgpu_{}Ú)r Ú device_optionZ device_typerÚCUDAÚformatZ device_id)rrrrÚ _name_scopeszCaffe2Rep._name_scopec sätt|ƒj|fi|¤Žt |jj¡Ft|tƒr‚t  |j ¡0|  ¡D]\}}|j   ||¡qJWdƒn1sv0Yn†t|tƒs–t|tƒrôt|jƒt|ƒkrÌtd t|jƒd |j¡t|ƒ¡ƒ‚t|ƒD]\}}|j   |j||¡qÔn|j   |jd|¡|js2|j  |j¡|j  |j¡d|_|jsP|j  |jj¡d|_|j  |jj¡Wdƒn1sv0Yg}|jjD]D}z| |j  |¡¡Wn&tyÌ| |j   |¡¡Yn0qŒt!d|jjƒ|ŽS)NzCExpected {} values for uninitialized graph inputs ({}), but got {}.z, rTZOutputs)"rrÚrunrZ DeviceScoper rÚ isinstanceÚdictZ NameScoperÚitemsr ZFeedBlobÚlistÚtupleÚlenr Ú RuntimeErrorrÚjoinÚ enumerater Z CreateNetr rZRunNetÚnameZexternal_outputÚappendZ FetchBlobÚ ExceptionZ FetchInt8Blobr)rÚinputsÚkwargsÚkeyÚvalueÚiZ output_valuesr#rrrrsF 0 ü0 ÿÿz Caffe2Rep.run)Ú__name__Ú __module__Ú __qualname__rÚpropertyrrÚ __classcell__rrrrr s  rN)Z caffe2.pythonrZ caffe2.protorZonnx.backend.baserrrrrrrÚs