Add count_include_pad to average_pool_gradient_op (#15997)
authorXiaomeng Yang <yangxm@fb.com>
Wed, 16 Jan 2019 00:44:33 +0000 (16:44 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Wed, 16 Jan 2019 00:56:40 +0000 (16:56 -0800)
commit13f38ab79d7ed781182c623cd5428b64c42f6254
treefbbc1c001d0203d7cb5f6a9112f1960581cd388b
parentb2eb98f6c371cca2689908d66d9771a26c69f90a
Add count_include_pad to average_pool_gradient_op (#15997)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15997

Add count_include_pad to average_pool_gradient_op

Reviewed By: houseroad

Differential Revision: D13648339

fbshipit-source-id: 205cb2acb32dc24a85256b628298b1a11f0ffa2c
caffe2/operators/pool_gradient_op.cc
caffe2/operators/pool_op.cu
caffe2/operators/pool_op.h
caffe2/python/operator_test/pooling_test.py