(self, input_param, input_shape, expected_shape)
| 32 | |
| 33 | @parameterized.expand(TESTS) |
| 34 | def test_pad(self, input_param, input_shape, expected_shape): |
| 35 | modes = ["constant", NumpyPadMode.CONSTANT, PytorchPadMode.CONSTANT] |
| 36 | self.pad_test(input_param, input_shape, expected_shape, modes) |
| 37 | |
| 38 | def test_pad_kwargs(self): |
| 39 | kwargs = {"spatial_border": 2, "mode": "constant"} |