projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 9d01be1) | patch
Remove second output of Reshape during ONNXIFI transform (#17027)
authorYinghai Lu <yinghai@fb.com>
Wed, 13 Feb 2019 02:21:09 +0000 (18:21 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Wed, 13 Feb 2019 02:31:53 +0000 (18:31 -0800)
commitfb5790ce94ec4359eca0aae061b73dc13b5aa4ec
tree2848377805c4d586334556165f09be6ec9fee93btree | snapshot
parent9d01be1a5aa347fc28ff019747e087844b3239b6commit | diff
Remove second output of Reshape during ONNXIFI transform (#17027)

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

Glow doesn't support second output of Reshape right now and it's useless. For correctness, we do make sure that the second output of Reshape is of Constant type during bound shape inference.

Reviewed By: ipiszy

Differential Revision: D14056555

fbshipit-source-id: f39cca7ba941bf5a5cc3adc96e2b1f943cc0be93
caffe2/opt/bound_shape_inference_test.cc diff | blob | history
caffe2/opt/bound_shape_inferencer.cc diff | blob | history
caffe2/opt/bound_shape_inferencer.h diff | blob | history
caffe2/opt/onnxifi_transformer.cc diff | blob | history
Domain: Machine Learning / ML Framework;