Searched refs:test_conv2d_valid_padding (Results 1 – 3 of 3) sorted by relevance
/aosp_15_r20/external/pytorch/test/xpu/ | ||
H A D | test_conv.py | 427 def test_conv2d_valid_padding(self, device, dtype): member in TestConvolutionNNDeviceType |
/aosp_15_r20/external/pytorch/test/nn/ | ||
H A D | test_convolution.py | 1780 def test_conv2d_valid_padding(self, device, dtype): member in TestConvolutionNNDeviceType |
/aosp_15_r20/external/pytorch/test/ | ||
H A D | test_mps.py | 8943 def test_conv2d_valid_padding(self, device='mps'): member in TestNNMPS |