projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 104773c) | patch
Blacklist certain op types when doing bound shape inference (#18290)
authorYinghai Lu <yinghai@fb.com>
Thu, 21 Mar 2019 22:28:20 +0000 (15:28 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 21 Mar 2019 22:43:05 +0000 (15:43 -0700)
commit979db037221d91f9dbd08df58a261d7448e6240d
tree70c11f8c170dc2e29b41499d78e59e9d98a74366tree | snapshot
parent104773c715acd79909bd7ef90376c61c25839793commit | diff
Blacklist certain op types when doing bound shape inference (#18290)

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

Some such as `Tile` will mess up our tracking of batch size and for now it makes sense to stop the shape inference on these ops so that we don't lower it and downstream ops without proper batch info.

Reviewed By: zrphercule

Differential Revision: D14463550

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