Searched defs:weight_memory_format (Results 1 – 2 of 2) sorted by relevance
329 auto weight_memory_format = weight.suggest_memory_format(); in cudnn_conv_suggest_memory_format() local371 auto weight_memory_format = weight.suggest_memory_format(); in miopen_conv_use_channels_last() local397 auto weight_memory_format = weight.suggest_memory_format(); in mkldnn_conv_use_channels_last() local413 auto weight_memory_format = weight.suggest_memory_format(); in thnn_conv_use_channels_last() local436 auto weight_memory_format = weight.suggest_memory_format(); in xpu_conv_use_channels_last() local
353 def _test_conv_deconv_nhwc_base(self, conv_module, weight_memory_format, dtype, prec=None): argument