projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 4171ef3) | patch
Add count_include_pad for avg_pool on CuDNN (#16100)
authorXiaomeng Yang <yangxm@fb.com>
Thu, 17 Jan 2019 10:07:04 +0000 (02:07 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 17 Jan 2019 10:10:12 +0000 (02:10 -0800)
commit7536887cb7fbcbaf964fcef979e28b693ec70106
treef11b101edc288b8b6cc27d638ba01bf1070e432dtree | snapshot
parent4171ef3728b92e50e5f791ab84c304994238dda0commit | diff
Add count_include_pad for avg_pool on CuDNN (#16100)

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

Add count_include_pad for avg_pool on CuDNN

Reviewed By: houseroad

Differential Revision: D13707959

fbshipit-source-id: 261f5d116066fef75cf9a5787dfbc5d12b5b9f9b
caffe2/operators/pool_op.cc diff | blob | history
caffe2/operators/pool_op.cu diff | blob | history
caffe2/operators/pool_op.h diff | blob | history
caffe2/operators/pool_op_cudnn.cc [new file with mode: 0644] blob
caffe2/operators/pool_op_cudnn.cu [deleted file] blob | history
caffe2/python/operator_test/pooling_test.py diff | blob | history
Domain: Machine Learning / ML Framework;