OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
3803d1c901
pytorch/caffe2/onnx
History
Rui Zhu 2f73b3d26e Add if ops support for onnxifi and ssa-rewrite (#19585) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/19585

Originally we will unroll all If op to many different subnets;
Now we will not unroll it anymore, but just add all external input of its subnet to the If op, and ssa-rewrite all external input/outputs. That would be enough.

Reviewed By: yinghai

Differential Revision: D15038139

fbshipit-source-id: 8532216d8749068acd5558ad0d8cb1d98463a063
2019-04-24 11:01:13 -07:00
..
torch_ops Add onnxifi support to SparseLengthsWeightedSum (#14210) 2018-11-21 15:47:24 -08:00
backend_rep.cc Update mobile predictor caller's interface 2018-08-24 23:40:05 -07:00
backend_rep.h Replace direct include of caffe2.pb.h with an intermediary header caffe2_pb.h (#10946) 2018-08-28 11:57:08 -07:00
backend.cc Resubmit PR-18512: Improved onnx export for 3 onnx ops (#18571) 2019-03-28 18:12:49 -07:00
backend.h Resubmit PR-18512: Improved onnx export for 3 onnx ops (#18571) 2019-03-28 18:12:49 -07:00
CMakeLists.txt [Caffe2] C++ SSA Rewrite of Caffe2 nets (#6531) 2018-04-17 14:24:13 -07:00
device.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
device.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
helper.cc Export complete subgraph io info when calling onnxGetBackendCompatibility (#14827) 2018-12-05 23:52:06 -08:00
helper.h Export complete subgraph io info when calling onnxGetBackendCompatibility (#14827) 2018-12-05 23:52:06 -08:00
onnx_exporter.cc Add if ops support for onnxifi and ssa-rewrite (#19585) 2019-04-24 11:01:13 -07:00
onnx_exporter.h Do not rename net boundary inputs/outputs during ssaRewrite. (#17545) 2019-03-06 14:26:58 -08:00
onnxifi_graph_info.cc Improve onnxifi backend init time (#17375) 2019-02-22 16:58:30 -08:00
onnxifi_graph_info.h Add foxi submodule (ONNXIFI facebook extension) 2019-02-25 08:00:03 -08:00
onnxifi_init.cc Enable single graph sharing between multiple threads for onnxifiop (#16047) 2019-01-16 12:19:16 -08:00
onnxifi_init.h Add foxi submodule (ONNXIFI facebook extension) 2019-02-25 08:00:03 -08:00
ssa_test.cc Do not rename net boundary inputs/outputs during ssaRewrite. (#17545) 2019-03-06 14:26:58 -08:00