Support conversion from Caffe2 MergeDim to ONNX Reshape + Squeeze. (#16189)
authorTongliang Liao <xkszltl@gmail.com>
Wed, 13 Feb 2019 22:57:27 +0000 (14:57 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Wed, 13 Feb 2019 23:53:38 +0000 (15:53 -0800)
commit491f2d4cb86da3cfaf908f538be20889e64bb4fa
treeeeee5cc5b80e45aa9d0fc9be8b554361b6c4057a
parent86594e63eb0419e5f6fe2ddb72ee707f4a4afdd5
Support conversion from Caffe2 MergeDim to ONNX Reshape + Squeeze. (#16189)

Summary:
`MergeDim` can be done by `Reshape([1, -1, 0, 0, ...]) + Squeeze`.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/16189

Differential Revision: D14070676

Pulled By: ezyang

fbshipit-source-id: 28d7e9b35cc2c1dcbd4afb3fbdf7383e219b1777
caffe2/onnx/onnx_exporter.cc
caffe2/onnx/onnx_exporter.h
caffe2/operators/prepend_dim_op.cc
caffe2/python/onnx/tests/c2_ref_test.py