projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: aff0964) | patch
Port the backend of FractionalMaxPool3d from TH to ATen (#15575)
authorChandler Zuo <chandlerzuo@fb.com>
Wed, 16 Jan 2019 22:01:39 +0000 (14:01 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Wed, 16 Jan 2019 22:16:30 +0000 (14:16 -0800)
commit237c0c3c7a551c6b27c5108ce30769bd41a542be
treebf6abe0adf2f11d4f567aef9263e2cfd6b40e8a8tree | snapshot
parentaff0964ee78215cf11bbd46883159c1f8694a3decommit | diff
Port the backend of FractionalMaxPool3d from TH to ATen (#15575)

Summary:
1. Port the FractionalMaxPool3d implementation from THNN/THCUNN to ATen.
2. Expose this function to Python module nn.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15575

Differential Revision: D13612848

Pulled By: chandlerzuo

fbshipit-source-id: 5f474b39005efa7788e984e8a805456dcdc43f6c
14 files changed:
aten/src/ATen/native/FractionalMaxPool3d.cpp [new file with mode: 0644] blob
aten/src/ATen/native/cuda/FractionalMaxPool3d.cu [new file with mode: 0644] blob
aten/src/ATen/native/native_functions.yaml diff | blob | history
aten/src/THCUNN/CMakeLists.txt diff | blob | history
aten/src/THCUNN/VolumetricFractionalMaxPooling.cu [deleted file] blob | history
aten/src/THCUNN/generic/THCUNN.h diff | blob | history
aten/src/THCUNN/generic/VolumetricFractionalMaxPooling.cu [deleted file] blob | history
aten/src/THNN/generic/VolumetricFractionalMaxPooling.c [deleted file] blob | history
aten/src/THNN/init.cpp diff | blob | history
test/common_nn.py diff | blob | history
tools/autograd/derivatives.yaml diff | blob | history
torch/nn/functional.py diff | blob | history
torch/nn/modules/__init__.py diff | blob | history
torch/nn/modules/pooling.py diff | blob | history
Domain: Machine Learning / ML Framework;