Searched refs:path_to_exported_script_module (Results 1 – 3 of 3) sorted by relevance
97 const std::string& path_to_exported_script_module) { in load_serialized_module_with_custom_op_and_execute() argument99 torch::jit::load(path_to_exported_script_module); in load_serialized_module_with_custom_op_and_execute()108 const std::string& path_to_exported_script_module) { in test_argument_checking_for_serialized_modules() argument110 torch::jit::load(path_to_exported_script_module); in test_argument_checking_for_serialized_modules()142 void test_move_to_device(const std::string& path_to_exported_script_module) { in test_move_to_device() argument144 torch::jit::load(path_to_exported_script_module); in test_move_to_device()163 void test_move_to_dtype(const std::string& path_to_exported_script_module) { in test_move_to_dtype() argument165 torch::jit::load(path_to_exported_script_module); in test_move_to_dtype()185 const std::string path_to_exported_script_module = argv[1]; in main() local191 path_to_exported_script_module); in main()[all …]
11 const std::string &path_to_exported_script_module) { in test_module_forward_invocation_no_hooks_run() argument15 torch::jit::load(path_to_exported_script_module + "_" + in test_module_forward_invocation_no_hooks_run()31 const std::string &path_to_exported_script_module) { in test_submodule_called_directly_with_hooks() argument35 torch::jit::load(path_to_exported_script_module + "_" + in test_submodule_called_directly_with_hooks()61 const std::string path_to_exported_script_module = argv[1]; in main() local62 std::cout << "path to exported module:" << path_to_exported_script_module in main()122 path_to_exported_script_module + "_" + test_case.name + ".pt"); in main()130 test_module_forward_invocation_no_hooks_run(path_to_exported_script_module); in main()131 test_submodule_called_directly_with_hooks(path_to_exported_script_module); in main()
32 const std::string path_to_exported_script_module = argv[1]; in main() local35 load_serialized_lowered_module_and_execute(path_to_exported_script_module); in main()