Searched refs:cpp_tmp_folder (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/test/cpp_api_parity/ |
| H A D | module_impl_check.py | 159 cpp_tmp_folder = test_params.cpp_tmp_folder
161 try_remove_folder(cpp_tmp_folder)
162 os.mkdir(cpp_tmp_folder)
171 cpp_tmp_folder, module_variant_name, "module"
174 cpp_tmp_folder, module_variant_name, "arg_dict"
184 cpp_tmp_folder, module_variant_name, "forward_output"
187 cpp_tmp_folder, module_variant_name, "backward_grad_dict"
248 try_remove_folder(cpp_tmp_folder)
289 cpp_tmp_folder=tempfile.mkdtemp(),
319 try_remove_folder(test_params.cpp_tmp_folder)
|
| H A D | functional_impl_check.py | 102 cpp_tmp_folder = test_params.cpp_tmp_folder
104 try_remove_folder(cpp_tmp_folder)
105 os.mkdir(cpp_tmp_folder)
112 cpp_tmp_folder, functional_variant_name, "arg_dict"
123 cpp_tmp_folder, functional_variant_name, "forward_output"
139 try_remove_folder(cpp_tmp_folder)
206 cpp_tmp_folder=tempfile.mkdtemp(),
253 try_remove_folder(test_params.cpp_tmp_folder)
|
| H A D | utils.py | 143 def compute_temp_file_path(cpp_tmp_folder, variant_name, file_suffix): argument
144 return os.path.join(cpp_tmp_folder, f"{variant_name}_{file_suffix}.pt")
|