Add support of count_include_pad and test end to end test for AveragePool (#17034)
authorDwarak Rajagopal <dwarak@fb.com>
Thu, 14 Feb 2019 18:28:25 +0000 (10:28 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 14 Feb 2019 19:48:42 +0000 (11:48 -0800)
commit65d6f1014a3689bef90cdc264e45de560f64b741
treee4b9955a1b45cebb78719b2dcb42fb334b451243
parent19addc7eb0bbb0074a3a57b6598382aeaa2222c9
Add support of count_include_pad and test end to end test for AveragePool (#17034)

Summary:
Add support of count_include_pad end to end test for AveragePool

We can export AveragePool from PyTorch with count_include_pad attribute. However, we don't directly support it in Caffe2's ONNX backend.
We also want to check whether we can pass the end to end test for average pool operator with count_include_pad attribute (pytorch => onnx => caffe2)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/17034

Reviewed By: houseroad

Differential Revision: D14060186

Pulled By: dwarakrajagopal

fbshipit-source-id: 10dae532611c71f8c8cfc3fa701cc7c1c1c02695
caffe2/onnx/backend.cc
test/onnx/test_pytorch_onnx_caffe2.py