projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 83c054d) | patch
Add global pooling specialization and also update MaxPooling on GPU (#15824)
authorXiaomeng Yang <yangxm@fb.com>
Sat, 12 Jan 2019 06:35:12 +0000 (22:35 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Sat, 12 Jan 2019 06:37:48 +0000 (22:37 -0800)
commit586d0303113123f59c1992c57714bec60b90fd76
tree2bc589a0e36633bfafa2fe4def057eac09097f7dtree | snapshot
parent83c054de481d4f65a8a73a903edd6beaac18e8bccommit | diff
Add global pooling specialization and also update MaxPooling on GPU (#15824)

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

Add global pooling specialization and also update MaxPooling on GPU

Reviewed By: houseroad

Differential Revision: D13596340

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