OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 37 Packages Projects Releases Wiki Activity
7103d0d938
pytorch/caffe2/python/onnx
History
Wanchao Liang 8aa23907e8 Make if block also take control_inputs, preserve SSA (#12224) 
Summary:
If block is missing control inputs when do caffe2 net execution, this PR add them back and remove the un-SSA semantics

jamesr66a
Pull Request resolved: https://github.com/pytorch/pytorch/pull/12224

Differential Revision: D10135408

Pulled By: wanchaol

fbshipit-source-id: 746c870bde54ed4ca627167361db1b3f36cd235c
2018-10-03 14:29:01 -07:00
..
bin Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
tests Make if block also take control_inputs, preserve SSA (#12224) 2018-10-03 14:29:01 -07:00
__init__.py Move onnx-caffe2 inside caffe2 (#1921) 2018-02-20 13:56:52 -08:00
backend_cpp_rep.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
backend_rep.py Back out "Revert D10123245: Back out "codemod cuda_gpu_id to device_id"" (#12232) 2018-10-01 21:54:52 -07:00
backend.py Make if block also take control_inputs, preserve SSA (#12224) 2018-10-03 14:29:01 -07:00
error.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
frontend.py Bump up the frontend version (#11873) 2018-09-20 16:20:48 -07:00
helper.py [Caffe2] Scoped dummy name generator (#6458) 2018-04-16 11:58:02 -07:00
onnxifi.py Support new Reshape semantics (#10848) 2018-08-24 11:46:41 -07:00
ONNXOpCoverage.md Update the ONNX op coverage in C2 2018-06-29 17:25:19 -07:00
README.md Add README and ONNXOpCoverage doc back (#2102) 2018-03-01 17:05:25 -08:00
test_onnxifi.py Remove weight from input of onnxifi backend op (#10575) 2018-08-16 10:56:25 -07:00
workspace.py Disallow using the OOP api workspace as context managers (#6456) 2018-04-09 22:13:54 -07:00

README.md

Caffe2 implementation of Open Neural Network Exchange (ONNX)

Usage

Installation

onnx-caffe2 is installed as a part of Caffe2. Please follow the instructions to install Caffe2.

Folder Structure

  • ./: the main folder that all code lies under
    • frontend.py: translate from caffe2 model to onnx model
    • backend.py: execution engine that runs onnx on caffe2
  • tests/: test files

Testing

onnx-caffe2 uses pytest as test driver. In order to run tests, first you need to install pytest:

pip install pytest-cov

After installing pytest, do

pytest

to run tests.

Testing coverage issues/status: https://github.com/caffe2/caffe2/blob/master/caffe2/python/onnx/ONNXOpCoverage.md

Development

During development it's convenient to install caffe2 in development mode:

cd /path/to/caffe2
pip install -e caffe2/

License

MIT License