Export ReduceMean/ReduceFrontMean/ReduceBackMean (Caffe2) to ReduceMean (ONNX). ...
authorTongliang Liao <xkszltl@gmail.com>
Tue, 12 Feb 2019 21:18:13 +0000 (13:18 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Tue, 12 Feb 2019 21:35:32 +0000 (13:35 -0800)
commit0eee56fff7c5da9261a70328f089c4172b8c1367
tree82f60bd31720e711bdf1a3ccaed2a0bf8fec4f8e
parentb0d57aa7b1de88106f19ad068510bb2e3eb26ffe
Export ReduceMean/ReduceFrontMean/ReduceBackMean (Caffe2) to ReduceMean (ONNX). (#16727)

Summary:
The second input (`lengths`) is not supported.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/16727

Differential Revision: D14054105

Pulled By: houseroad

fbshipit-source-id: 36b8d00460f9623696439e1bd2a6bc60b7bb263c
caffe2/onnx/onnx_exporter.cc
caffe2/onnx/onnx_exporter.h
caffe2/operators/reduce_front_back_mean_ops.cc
caffe2/operators/reduce_ops.cc
caffe2/python/onnx/tests/c2_ref_test.py