Searched refs:test_conv_double_backward (Results 1 – 3 of 3) sorted by relevance
/aosp_15_r20/external/pytorch/test/xpu/ | ||
H A D | test_conv.py | 288 def test_conv_double_backward(self, device, dtype): member in TestConvolutionNNDeviceType |
/aosp_15_r20/external/pytorch/test/nn/ | ||
H A D | test_convolution.py | 1527 def test_conv_double_backward(self, device, dtype): member in TestConvolutionNNDeviceType |
/aosp_15_r20/external/pytorch/test/functorch/ | ||
H A D | test_vmap.py | 4768 def test_conv_double_backward(self, device): member in TestVmapOperatorsOpInfo |