projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 0a5de6e) | patch
Remove fake inference for shape info in ONNXIFI transform (#17046)
authorYinghai Lu <yinghai@fb.com>
Thu, 14 Feb 2019 22:22:51 +0000 (14:22 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 14 Feb 2019 23:12:20 +0000 (15:12 -0800)
commitb515ebc6f11400d904339275aa5df12f1f8121b1
tree9696946487cdac7729b5359de2d0013173f6892ftree | snapshot
parent0a5de6e9720eb9600c6424fd2d7a5df0b36d9703commit | diff
Remove fake inference for shape info in ONNXIFI transform (#17046)

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

As we are moving to use bound shape inference, we can remove the awkward fake inference run path and make the code cleaner.

Reviewed By: ipiszy

Differential Revision: D14061501

fbshipit-source-id: b3ace98b3dabef3c3359086a0bb1410518cefa26
caffe2/opt/onnxifi_transformer.cc diff | blob | history
caffe2/opt/onnxifi_transformer.h diff | blob | history
caffe2/python/onnx/onnxifi.py diff | blob | history
caffe2/python/pybind_state.cc diff | blob | history
Domain: Machine Learning / ML Framework;