Searched refs:backend_memory_format (Results 1 – 2 of 2) sorted by relevance
1416 at::MemoryFormat backend_memory_format = at::MemoryFormat::Contiguous; in determine_backend_memory_format() local1424 backend_memory_format = cudnn_conv_suggest_memory_format(input, weight); in determine_backend_memory_format()1433 …backend_memory_format = (k == 5) ? at::MemoryFormat::Contiguous /*at::MemoryFormat::ChannelsLast3d… in determine_backend_memory_format()1439 …backend_memory_format = (k == 5) ? at::MemoryFormat::ChannelsLast3d : at::MemoryFormat::ChannelsLa… in determine_backend_memory_format()1446 backend_memory_format = at::MemoryFormat::ChannelsLast; in determine_backend_memory_format()1451 …backend_memory_format = (k == 5) ? at::MemoryFormat::ChannelsLast3d : at::MemoryFormat::ChannelsLa… in determine_backend_memory_format()1455 backend_memory_format = at::MemoryFormat::Contiguous; in determine_backend_memory_format()1458 return backend_memory_format; in determine_backend_memory_format()1516 at::MemoryFormat backend_memory_format = determine_backend_memory_format(input, weight, backend); in _convolution() local1533 input.contiguous(backend_memory_format), weight, params.padding, params.stride, in _convolution()[all …]
578 at::MemoryFormat backend_memory_format = at::MemoryFormat::Contiguous; in convolution_overrideable() local580 …backend_memory_format = (k == 5) ? at::MemoryFormat::ChannelsLast3d : at::MemoryFormat::ChannelsLa… in convolution_overrideable()582 Tensor input_c = input_r.contiguous(backend_memory_format); in convolution_overrideable()583 Tensor weight_c = weight_r.contiguous(backend_memory_format); in convolution_overrideable()