projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 99d2743) | patch
Add count_include_pad arg for AveragePoolOp on GPU (#15787)
authorXiaomeng Yang <yangxm@fb.com>
Tue, 8 Jan 2019 05:33:44 +0000 (21:33 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Tue, 8 Jan 2019 05:36:26 +0000 (21:36 -0800)
commit4650d70e930fd90484e648bd7a7c6b7ac2bafb72
tree76322ffd20ed66062547311131b706282caa17dftree | snapshot
parent99d274386353e99d1568448d6e798d216e176d21commit | diff
Add count_include_pad arg for AveragePoolOp on GPU (#15787)

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

Add count_include_pad arg for AveragePoolOp on GPU

Reviewed By: houseroad

Differential Revision: D13589185

fbshipit-source-id: 235a84cfcd2033ee796c13e338fc3d03e832b5b1
caffe2/operators/pool_gradient_op.cc diff | blob | history
caffe2/operators/pool_op.cu diff | blob | history
Domain: Machine Learning / ML Framework;