/aosp_15_r20/external/pytorch/torch/utils/ |
H A D | mobile_optimizer.py | 16 script_module: torch.jit.ScriptModule, 33 if not isinstance(script_module, torch.jit.ScriptModule): 48 …bundled_inputs_attributes = _get_bundled_inputs_preserved_attributes(script_module, preserved_meth… 49 if all(hasattr(script_module, method) for method in bundled_inputs_attributes): 54 if not hasattr(script_module, method): 63 script_module._c, 68 script_module._c, 72 …optimized_cpp_module = torch._C._jit_pass_metal_optimize_for_mobile(script_module._c, preserved_me… 79 def generate_mobile_module_lints(script_module: torch.jit.ScriptModule): 89 if not isinstance(script_module, torch.jit.ScriptModule): [all …]
|
H A D | bundled_inputs.py | 412 def _get_bundled_inputs_attributes_and_methods(script_module: torch.jit.ScriptModule) -> Tuple[List… 417 if hasattr(script_module, 'get_all_bundled_inputs'): 422 if hasattr(script_module, 'get_bundled_inputs_functions_and_info'): 424 all_info = script_module.get_bundled_inputs_functions_and_info() 431 script_module, 437 func = getattr(script_module, function_name) 446 if hasattr(script_module, helper_fn_name):
|
/aosp_15_r20/external/pytorch/test/mobile/ |
H A D | test_lite_script_type.py | 26 script_module = torch.jit.script(MyTestModule()) 27 script_module_result = script_module(sample_input).a 30 script_module._save_to_buffer_for_lite_interpreter( 89 script_module = torch.jit.script(Bar()) 91 script_module_result = script_module(sample_input) 93 buffer_mobile = io.BytesIO(script_module._save_to_buffer_for_lite_interpreter()) 113 script_module = torch.jit.script(Bar()) 114 script_module_result = script_module(sample_input) 116 buffer_mobile = io.BytesIO(script_module._save_to_buffer_for_lite_interpreter()) 130 script_module = torch.jit.script(MyTestModule()) [all …]
|
H A D | test_lite_script_module.py | 51 script_module = torch.jit.script(MyTestModule()) 52 script_module_result = script_module(input) 54 buffer = io.BytesIO(script_module._save_to_buffer_for_lite_interpreter()) 124 script_module = torch.jit.script(MyTestModule()) 125 script_module_result = script_module(input) 128 script_module._save_to_buffer_for_lite_interpreter( 153 script_module = torch.jit.script(MyTestModule()) 154 script_module_result = script_module(*input) 156 buffer = io.BytesIO(script_module._save_to_buffer_for_lite_interpreter()) 164 script_module, [input], [] [all …]
|
H A D | test_bytecode.py | 365 script_module = torch.jit.script(MyTestModule()) 366 script_module_result = script_module(sample_input) 368 buffer = io.BytesIO(script_module._save_to_buffer_for_lite_interpreter())
|
H A D | test_upgraders.py | 17 def _save_load_mobile_module(self, script_module: torch.jit.ScriptModule): 19 script_module._save_to_buffer_for_lite_interpreter(
|
/aosp_15_r20/external/pytorch/torch/jit/ |
H A D | _recursive.py | 579 def init_fn(script_module): argument 611 script_module._modules[name] = scripted 621 bound_method = unbound_function.__get__(script_module) 622 setattr(script_module, name, bound_method) 624 setattr(script_module, name, item) 627 script_module._concrete_type = concrete_type 630 script_module = torch.jit.RecursiveScriptModule._construct(cpp_module, init_fn) 645 for idx, fn in enumerate(script_module._c._get_forward_pre_hooks()): 646 script_module._forward_pre_hooks[idx] = fn 647 for idx, fn in enumerate(script_module._c._get_forward_hooks()): [all …]
|
H A D | _script.py | 649 script_module = RecursiveScriptModule(cpp_module) 650 init_fn(script_module) 654 RecursiveScriptModule._finalize_scriptmodule(script_module) 655 return script_module 658 def _finalize_scriptmodule(script_module): argument 659 script_module._parameters = OrderedDictWrapper( 660 torch._C.ParameterDict(script_module._c) 662 script_module._buffers = OrderedDictWrapper( 663 torch._C.BufferDict(script_module._c) 665 script_module._modules = OrderedModuleDict( [all …]
|
H A D | _trace.py | 1397 script_module = torch.jit._recursive.create_script_module( 1402 self.__dict__["_actual_script_module"] = script_module
|
H A D | _script.pyi | 152 def _finalize_scriptmodule(script_module) -> None: ...
|
/aosp_15_r20/external/pytorch/test/jit/fixtures_srcs/ |
H A D | generate_models.py | 146 def get_operator_list(script_module: torch) -> Set[str]: 147 buffer = io.BytesIO(script_module._save_to_buffer_for_lite_interpreter()) 159 def get_output_model_version(script_module: torch.nn.Module) -> int: 161 torch.jit.save(script_module, buffer) 220 script_module = torch.jit.script(a_module) 221 actual_model_version = get_output_model_version(script_module) 234 actual_operator_list = get_operator_list(script_module) 246 script_module._save_for_lite_interpreter(export_model_path)
|
/aosp_15_r20/external/pytorch/test/cpp_api_parity/ |
H A D | module_impl_check.py | 136 script_module = torch.jit.trace(module, torch.tensor(0)) 154 return script_module, python_output, python_grad_dict 165 script_module, python_output, python_grad_dict = run_python_forward_backward( 176 script_module.save(module_file_path)
|
/aosp_15_r20/external/pytorch/android/pytorch_android/ |
H A D | generate_test_torchscripts.py | 12 script_module = torch.jit.script(module) 13 print(script_module.graph) 16 script_module._save_for_lite_interpreter(outputFileName)
|
/aosp_15_r20/external/pytorch/test/cpp/aoti_inference/ |
H A D | compile_model.py | 72 script_module = make_script_module(lib_path, device, *inputs) 74 script_module.save(aoti_script_model)
|
/aosp_15_r20/external/pytorch/torch/backends/_coreml/ |
H A D | preprocess.py | 86 def preprocess(script_module: torch._C.ScriptObject, compile_spec: Dict[str, Tuple]): 104 model = torch.jit.RecursiveScriptModule._construct(script_module, lambda x: None)
|
/aosp_15_r20/external/pytorch/torch/csrc/utils/ |
H A D | throughput_benchmark.cpp | 43 ThroughputBenchmark::ThroughputBenchmark(const jit::Module& script_module) in ThroughputBenchmark() argument 44 : script_module_(script_module) {} in ThroughputBenchmark()
|
/aosp_15_r20/external/pytorch/test/jit/ |
H A D | test_generator.py | 164 script_module = torch.jit.script(Foo()) 166 torch.jit.save(script_module, saved_module)
|
H A D | test_save_load.py | 413 script_module = torch.jit.script(Foo()) 415 script_module.save("NonExist/path/test.pt") 755 def script_module_to_buffer(script_module): argument 757 script_module._save_to_buffer_for_lite_interpreter(_use_flatbuffer=True) 1189 script_module = torch.jit.script(module) 1192 script_module_io = script_module._save_to_buffer_for_lite_interpreter(
|
H A D | test_class_type.py | 1109 script_module = torch.jit.script(ModuleWithUnused()) 1112 self.assertEqual(python_module.forward(), script_module.forward()) 1116 script_module.calls_unused() 1119 script_module.calls_unused_indirectly()
|
/aosp_15_r20/external/pytorch/torch/distributed/rpc/ |
H A D | internal.py | 87 def _script_module_reducer(self, script_module): argument 92 torch.jit.save(script_module, f)
|
/aosp_15_r20/external/pytorch/torch/testing/_internal/ |
H A D | jit_metaprogramming_utils.py | 591 def script_module(*args, **kwargs): function 625 return script_module
|
H A D | common_quantization.py | 1198 script_module = torch.jit.script(model) 1199 script_module_result = script_module(input) 1205 buffer = io.BytesIO(script_module._save_to_buffer_for_lite_interpreter())
|
/aosp_15_r20/external/pytorch/test/ |
H A D | test_jit.py | 311 def all_backward_graphs(script_module, diff_graph_idx=None): argument 313 ge_state = script_module.get_debug_state() 320 def backward_graph(script_module, diff_graph_idx=None, skip_check=False): argument 321 ge_state = script_module.get_debug_state()
|