Searched refs:test_grid_sample (Results 1 – 5 of 5) sorted by relevance
| /external/pytorch/test/quantization/core/ | ||
| D | test_quantized_functional.py | 232 def test_grid_sample(self, N, C, H, H_out, W, W_out, scale, zero_point): member in TestQuantizedFunctionalOps |
| /external/pytorch/test/onnx/ | ||
| D | test_onnx_opset.py | 558 def test_grid_sample(self): member in TestONNXOpset |
| D | test_pytorch_onnx_onnxruntime.py | 13555 def test_grid_sample(self, mode, padding_mode, align_corners): member in TestONNXRuntime |
| /external/pytorch/test/ | ||
| D | test_nn.py | 5530 def test_grid_sample(self): member in TestNN |
| D | test_mps.py | 10605 def test_grid_sample(self): member in TestConvolutionMPS |