/usr/local/lib64/python3.6/site-packages/caffe2/python/onnx/__pycache__/
/usr/local/lib64/python3.6/site-packages/caffe2/python/onnx/__pycache__
NameSizeModeActions
backend.cpython-36.pyc289010644editdlrm
backend_cpp_rep.cpython-36.pyc18690644editdlrm
backend_rep.cpython-36.pyc20610644editdlrm
error.cpython-36.pyc4500644editdlrm
frontend.cpython-36.pyc103440644editdlrm
helper.cpython-36.pyc36290644editdlrm
onnxifi.cpython-36.pyc16310644editdlrm
test_onnxifi.cpython-36.pyc61310644editdlrm
workspace.cpython-36.pyc25900644editdlrm
__init__.cpython-36.pyc1540644editdlrm
Edit: /usr/local/lib64/python3.6/site-packages/caffe2/python/onnx/__pycache__/onnxifi.cpython-36.pyc (1631B)
3 EgÁã @s:dZddlmZddljjZdd„Zdd„Zd d d „Z dS)z ONNXIFI a Caffe2 net é)Ú caffe2_pb2NcCstj|t|ƒƒS)z Set onnxifi option )ÚCÚonnxifi_set_optionÚstr)Ú option_nameZ option_value©rúF/usr/local/lib64/python3.6/site-packages/caffe2/python/onnx/onnxifi.pyr srcCs tj|ƒS)z Get onnxifi option )rÚonnxifi_get_option)rrrrr sr éFTc  Csòtjƒ} t|ƒtjkr|} n„t|ƒtkr xj|jƒD]^\} }tjƒ}| |_|jjj |ƒ|j j tjj gt |jjƒƒtjj |j d<| jj |gƒq2W|| _|| _tj|jƒ| jƒ|r¸|ng| dk rÆ| ng||| |||| |ƒ }tjƒ}|j|ƒ|S)z[ Transform the caffe2_net by collapsing ONNXIFI-runnable nodes into Onnxifi c2 ops rN)rZTensorBoundShapesÚtypeÚdictÚitemsZTensorBoundShapeÚnameÚshapeZdimsÚextendZdim_typeZCONSTANTÚlenZBATCHZshapesÚmax_batch_sizeZmax_feature_lenrZonnxifiZSerializeToStringZNetDefZParseFromString)Zpred_netZ input_shapesrZ max_seq_sizeÚdebugZuse_onnxZmerge_fp32_inputs_into_fp16Z adjust_batchZ block_listZ weight_namesZnet_ssa_rewrittenÚtimeoutZ shape_hintsÚkÚvZtbsZ pred_net_strZ pred_net_cutrrrÚonnxifi_caffe2_nets8    r) r r FTFTNNFr) Ú__doc__Z caffe2.protorZ!caffe2.python._import_c_extensionÚpythonZ_import_c_extensionrrr rrrrrÚs