pytorch/caffe2/onnx
Hao Lu 53dff784e2 [caffe2] Fix inplace ops in onnx::SsaRewrite (#46134)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/46134

Make sure in-place ops stay in-place after SsaRewrite. This seems to break the premise of SSA, but it's necessary to ensure correctness. Note here we only preserve the inplace ops that enforce inplace. Ops like `Relu` don't enforce inplace, they allow inplace.

(Note: this ignores all push blocking failures!)

Reviewed By: yinghai

Differential Revision: D24234957

fbshipit-source-id: 274bd3ad6227fce6a98e615aad7e57cd2696aec3
2020-10-22 13:26:31 -07:00
..
torch_ops Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00
backend_rep.cc Replace c10::guts::stuff with std::stuff (#30915) 2019-12-16 13:57:19 -08:00
backend_rep.h [codemod][caffe2] Run clang-format - 5/7 2020-06-30 15:45:11 -07:00
backend.cc Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00
backend.h [codemod][caffe2] Run clang-format - 5/7 2020-06-30 15:45:11 -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 [codemod][caffe2] Run clang-format - 5/7 2020-06-30 15:45:11 -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
offline_tensor.cc Support Offline Tensors through ONNXIFI layer 2019-10-31 10:33:42 -07:00
offline_tensor.h Add python inferface to create OfflineTensor (#42516) 2020-08-04 01:31:34 -07:00
onnx_exporter.cc [caffe2] Fix inplace ops in onnx::SsaRewrite (#46134) 2020-10-22 13:26:31 -07:00
onnx_exporter.h [caffe2] Fix inplace ops in onnx::SsaRewrite (#46134) 2020-10-22 13:26:31 -07:00
onnxifi_graph_info.cc Improve onnxifi backend init time (#17375) 2019-02-22 16:58:30 -08:00
onnxifi_graph_info.h Save the weight shape info the first time we have chance to extract it (#21233) 2019-06-01 12:55:29 -07: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 [caffe2] Fix inplace ops in onnx::SsaRewrite (#46134) 2020-10-22 13:26:31 -07:00