projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: ba287ee) | patch
Export complete subgraph io info when calling onnxGetBackendCompatibility (#14827)
authorYinghai Lu <yinghai@fb.com>
Thu, 6 Dec 2018 07:50:12 +0000 (23:50 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 6 Dec 2018 07:52:06 +0000 (23:52 -0800)
commit39d50ef4f69051105a055f8fa7cc4a9cf27eec33
tree2cd0e82a08febcba417eacf08e18a4dece126041tree | snapshot
parentba287eebca61a54bd0e6ec57b2761abf1110ffddcommit | diff
Export complete subgraph io info when calling onnxGetBackendCompatibility (#14827)

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

We need to send complete IO info when doing `onnxGetBackendCompatibility` to backend like Glow. Previously we are missing some info because sometimes we generate more than one nodes from one C2 op. This fixes the issue.

Reviewed By: jackm321

Differential Revision: D13352049

fbshipit-source-id: 8d8ac70656a0ac42f3a0ccecad61456a4f3b2435
caffe2/onnx/helper.cc diff | blob | history
caffe2/onnx/helper.h diff | blob | history
caffe2/opt/onnxifi_transformer.cc diff | blob | history
Domain: Machine Learning / ML Framework;