pytorch/test/cpp_api_parity
Thomas J. Fan c16f87949f ENH Adds nn.ReflectionPad3d (#59791)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/27655

This PR adds a C++ and Python version of ReflectionPad3d with structured kernels. The implementation uses lambdas extensively to better share code from the backward and forward pass.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/59791

Reviewed By: gchanan

Differential Revision: D29242015

Pulled By: jbschlosser

fbshipit-source-id: 18e692d3b49b74082be09f373fc95fb7891e1b56
2021-06-21 10:53:14 -07:00
..
__init__.py
functional_impl_check.py
module_impl_check.py
parity-tracker.md ENH Adds nn.ReflectionPad3d (#59791) 2021-06-21 10:53:14 -07:00
parity_table_parser.py
sample_functional.py
sample_module.py
utils.py