Searched defs:conv_module (Results 1 – 3 of 3) 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): argument
353 def _test_conv_deconv_nhwc_base(self, conv_module, weight_memory_format, dtype, prec=None): argument
|
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | pre_grad.py | 377 conv_module, argument
|
| /aosp_15_r20/external/pytorch/test/quantization/core/ |
| H A D | test_quantized_module.py | 266 self, module_name, qconv_module, conv_module, batch_size, argument
|