Support communicating with C2 protobuf in Onnxifi flow (#15472)
authorYinghai Lu <yinghai@fb.com>
Tue, 8 Jan 2019 06:09:38 +0000 (22:09 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Tue, 8 Jan 2019 06:12:29 +0000 (22:12 -0800)
commitc3a0000864e78513f3a7d9b3bbab0e216b783be0
tree95eb49df57068058a2977bfc5732580e15a1a04d
parent4650d70e930fd90484e648bd7a7c6b7ac2bafb72
Support communicating with C2 protobuf in Onnxifi flow (#15472)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15472

Create a path to pass serialized C2 protobuf instead of ONNX during ONNXIFI flow

Reviewed By: houseroad

Differential Revision: D13536603

fbshipit-source-id: 7d016474f4beedbda480ed2e2c0004af7868aafe
caffe2/operators/onnxifi_op.h
caffe2/opt/onnxifi_transformer.cc
caffe2/opt/onnxifi_transformer.h
caffe2/python/onnx/onnxifi.py
caffe2/python/pybind_state.cc