Searched defs:input_memory_format (Results 1 – 3 of 3) sorted by relevance
328 auto input_memory_format = input.suggest_memory_format(); in cudnn_conv_suggest_memory_format() local370 auto input_memory_format = input.suggest_memory_format(); in miopen_conv_use_channels_last() local396 auto input_memory_format = input.suggest_memory_format(); in mkldnn_conv_use_channels_last() local412 auto input_memory_format = input.suggest_memory_format(); in thnn_conv_use_channels_last() local435 auto input_memory_format = input.suggest_memory_format(); in xpu_conv_use_channels_last() local
136 at::MemoryFormat input_memory_format; in create_inputs() local
187 at::MemoryFormat input_memory_format; in create_inputs() local