Home
last modified time | relevance | path

Searched refs:input_memory_format_list (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/binaries/
H A Dcompare_models_torch.cc147 std::vector<std::string> input_memory_format_list = in create_inputs() local
155 input_memory_format_list.size(), in create_inputs()
165 input_memory_format_list.size(), in create_inputs()
188 if (input_memory_format_list[i] == "channels_last") { in create_inputs()
194 } else if (input_memory_format_list[i] == "contiguous_format") { in create_inputs()
198 "Unsupported input memory format: ", input_memory_format_list[i]); in create_inputs()
H A Dspeed_benchmark_torch.cc105 std::vector<std::string> input_memory_format_list = in create_inputs() local
114 input_memory_format_list.size(), in create_inputs()
137 if (input_memory_format_list[i] == "channels_last") { in create_inputs()
143 } else if (input_memory_format_list[i] == "contiguous_format") { in create_inputs()
147 "Unsupported input memory format: ", input_memory_format_list[i]); in create_inputs()