Bicubic interpolation for nn.functional.interpolate (#9849)
authorDavid Riazati <davidriazati@fb.com>
Mon, 17 Dec 2018 23:22:07 +0000 (15:22 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Mon, 17 Dec 2018 23:31:48 +0000 (15:31 -0800)
commit59d71b9664b57b0ea0de0d87cea87b21daa4dd7b
tree8adb548deaabad91d0ed35b6c9560aa4de580298
parentc5dd91c4aeb6184dd1e11943385da72f75a1e2f1
Bicubic interpolation for nn.functional.interpolate (#9849)

Summary:
Addresses #918, interpolation results should be similar to tf

* Adds bicubic interpolation operator to `nn.functional.interpolate`
* Corresponding test in `test_nn.py`

The operator is added in legacy `TH` to be aligned with the other upsampling operators; they can be refactored/moved to ATen all at once when #10482 is resolved
Pull Request resolved: https://github.com/pytorch/pytorch/pull/9849

Differential Revision: D9007525

Pulled By: driazati

fbshipit-source-id: 93ef49a34ce4e5ffd4bda94cd9a6ddc939f0a4cc
37 files changed:
aten/src/ATen/core/aten_interned_strings.h
aten/src/ATen/native/LegacyNNDefinitions.cpp
aten/src/ATen/native/native_functions.yaml
aten/src/ATen/nn.yaml
aten/src/THCUNN/CMakeLists.txt
aten/src/THCUNN/SpatialUpSamplingBicubic.cu [new file with mode: 0644]
aten/src/THCUNN/SpatialUpSamplingBilinear.cu
aten/src/THCUNN/SpatialUpSamplingNearest.cu
aten/src/THCUNN/TemporalUpSamplingLinear.cu
aten/src/THCUNN/TemporalUpSamplingNearest.cu
aten/src/THCUNN/VolumetricUpSamplingNearest.cu
aten/src/THCUNN/VolumetricUpSamplingTrilinear.cu
aten/src/THCUNN/generic/SpatialUpSamplingBicubic.cu [new file with mode: 0644]
aten/src/THCUNN/generic/SpatialUpSamplingBilinear.cu
aten/src/THCUNN/generic/THCUNN.h
aten/src/THCUNN/generic/TemporalUpSamplingLinear.cu
aten/src/THCUNN/generic/VolumetricUpSamplingTrilinear.cu
aten/src/THCUNN/linear_upsampling.h [deleted file]
aten/src/THCUNN/upsampling.h [new file with mode: 0644]
aten/src/THNN/generic/SpatialUpSamplingBicubic.c [new file with mode: 0644]
aten/src/THNN/generic/SpatialUpSamplingBilinear.c
aten/src/THNN/generic/SpatialUpSamplingNearest.c
aten/src/THNN/generic/THNN.h
aten/src/THNN/generic/TemporalUpSamplingLinear.c
aten/src/THNN/generic/TemporalUpSamplingNearest.c
aten/src/THNN/generic/VolumetricUpSamplingNearest.c
aten/src/THNN/generic/VolumetricUpSamplingTrilinear.c
aten/src/THNN/generic/linear_upsampling.h [deleted file]
aten/src/THNN/generic/upsampling.h [new file with mode: 0644]
aten/src/THNN/init.cpp
test/common_nn.py
test/test_nn.py
tools/autograd/derivatives.yaml
torch/csrc/jit/register_prim_ops.cpp
torch/cuda/__init__.py
torch/nn/functional.py
torch/nn/modules/upsampling.py