Searched refs:weight_memory_format (Results 1 – 3 of 3) sorted by relevance
329 auto weight_memory_format = weight.suggest_memory_format(); in cudnn_conv_suggest_memory_format() local334 (weight_memory_format == at::MemoryFormat::ChannelsLast) in cudnn_conv_suggest_memory_format()342 (weight_memory_format == at::MemoryFormat::ChannelsLast3d) in cudnn_conv_suggest_memory_format()371 auto weight_memory_format = weight.suggest_memory_format(); in miopen_conv_use_channels_last() local375 (weight_memory_format == at::MemoryFormat::ChannelsLast) ) in miopen_conv_use_channels_last()397 auto weight_memory_format = weight.suggest_memory_format(); in mkldnn_conv_use_channels_last() local401 (weight_memory_format == at::MemoryFormat::ChannelsLast); in mkldnn_conv_use_channels_last()405 (weight_memory_format == at::MemoryFormat::ChannelsLast3d); in mkldnn_conv_use_channels_last()413 auto weight_memory_format = weight.suggest_memory_format(); in thnn_conv_use_channels_last() local417 weight_memory_format == at::MemoryFormat::ChannelsLast)); in thnn_conv_use_channels_last()[all …]
474 weight_memory_format = torch.contiguous_format488 conv.weight.data = conv.weight.to(memory_format=weight_memory_format)501 weight_memory_format = torch.contiguous_format514 conv.weight.data = conv.weight.to(memory_format=weight_memory_format)
353 def _test_conv_deconv_nhwc_base(self, conv_module, weight_memory_format, dtype, prec=None): argument380 conv2 = copy.deepcopy(conv1).to(memory_format=weight_memory_format)