Searched refs:_test_conv_deconv_lower_precision_base (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_mkldnn.py | 291 def _test_conv_deconv_lower_precision_base(self, dim, conv_module, dtype): member in TestMkldnn
340 self._test_conv_deconv_lower_precision_base(1, torch.nn.Conv1d, dtype=dtype)
341 self._test_conv_deconv_lower_precision_base(1, torch.nn.ConvTranspose1d, dtype=dtype)
345 self._test_conv_deconv_lower_precision_base(2, torch.nn.Conv2d, dtype=dtype)
346 self._test_conv_deconv_lower_precision_base(2, torch.nn.ConvTranspose2d, dtype=dtype)
350 self._test_conv_deconv_lower_precision_base(3, torch.nn.Conv3d, dtype=dtype)
351 self._test_conv_deconv_lower_precision_base(3, torch.nn.ConvTranspose3d, dtype=dtype)
|