Caffe 2: Reshape Op upgrade (#15380)
authorSergei Nikolaev <drnikolaev@users.noreply.github.com>
Mon, 14 Jan 2019 06:46:39 +0000 (22:46 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Mon, 14 Jan 2019 06:49:40 +0000 (22:49 -0800)
commita282378baf1a1e674ce5480d33173dbb577716c0
tree03a50d3ca84511a74a8b11d2b56029e5c5fe53d2
parent04b8a2f1baa87efc0434a532e036a0bb29de22a9
Caffe 2: Reshape Op upgrade (#15380)

Summary:
This is follow up on #13945 where we had to turn off some TRT tests because some ops were not ready to accept ONNX opset 9+ models. This PR fixes Reshape.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15380

Differential Revision: D13649825

Pulled By: houseroad

fbshipit-source-id: b72e62803de5b63cc001c3fe4b3bf64dfa996e94
caffe2/python/operator_test/reshape_ops_test.py
caffe2/python/trt/test_trt.py
caffe2/python/trt/transform.py