Method
test_shape
(self, input_param, input_shape, expected_shape)
Source from the content-addressed store, hash-verified
| 124 | |
| 125 | @parameterized.expand(TEST_CASES + TEST_CASES_EQ + TEST_CASES_EQ2) |
| 126 | def test_shape(self, input_param, input_shape, expected_shape): |
| 127 | net = UpSample(**input_param) |
| 128 | with eval_mode(net): |
| 129 | result = net(torch.randn(input_shape)) |
| 130 | self.assertEqual(result.shape, expected_shape, msg=str(input_param)) |
| 131 | |
| 132 | |
| 133 | if __name__ == "__main__": |
Callers
nothing calls this directly
Tested by
no test coverage detected