Home
last modified time | relevance | path

Searched refs:input_memory_format (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DConvUtils.h328 auto input_memory_format = input.suggest_memory_format(); in cudnn_conv_suggest_memory_format() local
333 (input_memory_format == at::MemoryFormat::ChannelsLast) || in cudnn_conv_suggest_memory_format()
341 (input_memory_format == at::MemoryFormat::ChannelsLast3d) || in cudnn_conv_suggest_memory_format()
370 auto input_memory_format = input.suggest_memory_format(); in miopen_conv_use_channels_last() local
374 ( (input_memory_format == at::MemoryFormat::ChannelsLast) || in miopen_conv_use_channels_last()
396 auto input_memory_format = input.suggest_memory_format(); in mkldnn_conv_use_channels_last() local
400 (input_memory_format == at::MemoryFormat::ChannelsLast) || in mkldnn_conv_use_channels_last()
404 (input_memory_format == at::MemoryFormat::ChannelsLast3d) || in mkldnn_conv_use_channels_last()
412 auto input_memory_format = input.suggest_memory_format(); in thnn_conv_use_channels_last() local
416 (input_memory_format == at::MemoryFormat::ChannelsLast) || ( in thnn_conv_use_channels_last()
[all …]
/aosp_15_r20/external/pytorch/binaries/
H A Dcompare_models_torch.cc48 input_memory_format,
187 at::MemoryFormat input_memory_format; in create_inputs() local
193 input_memory_format = at::MemoryFormat::ChannelsLast; in create_inputs()
195 input_memory_format = at::MemoryFormat::Contiguous; in create_inputs()
203 …at::TensorOptions(input_type).memory_format(input_memory_format))*(range_max - range_min) - range_… in create_inputs()
H A Dspeed_benchmark_torch.cc45 input_memory_format,
136 at::MemoryFormat input_memory_format; in create_inputs() local
142 input_memory_format = at::MemoryFormat::ChannelsLast; in create_inputs()
144 input_memory_format = at::MemoryFormat::Contiguous; in create_inputs()
154 memory_format(input_memory_format))); in create_inputs()
/aosp_15_r20/external/executorch/backends/apple/mps/test/
H A Dtest_mps.py473 input_memory_format = torch.contiguous_format
478 torch.randn(N, C, H, W).to(memory_format=input_memory_format),
500 input_memory_format = torch.contiguous_format
505 torch.randn(N, C, H, W).to(memory_format=input_memory_format),