Searched refs:input_memory_format (Results 1 – 4 of 4) sorted by relevance
/aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
H A D | ConvUtils.h | 328 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 D | compare_models_torch.cc | 48 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 D | speed_benchmark_torch.cc | 45 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 D | test_mps.py | 473 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),
|