pytorch
152c1529 - Add tests for all padding layers to `module_db` in `common_modules.py` (#96641)

Commit
1 year ago
Add tests for all padding layers to `module_db` in `common_modules.py` (#96641) Adding the PR discussed in #96295. - Adds tests for all current padding layers to `module_db` in `torch/testing/_internal/common_modules.py` ( `nn.ReflectionPad`, `nn.ReplicationPad`, `nn.ZeroPad`, `nn.ConstantPad` ) for 1D, 2D, and 3D variants. - Removes tests for the same padding layers from `torch/testing/_internal/common_nn.py` Pull Request resolved: https://github.com/pytorch/pytorch/pull/96641 Approved by: https://github.com/albanD
Committer
Parents
Loading