Use bound shape inference in onnxifi transform (#16598)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/16598
ATT.
Reviewed By: bertmaher, rdzhabarov
Differential Revision: D13893698
fbshipit-source-id: 8d2ad9814fe76924a46b450eb7ebd3601fbdbbc7