Home
last modified time | relevance | path

Searched refs:script_module (Results 1 – 23 of 23) sorted by relevance

/aosp_15_r20/external/pytorch/torch/utils/
H A Dmobile_optimizer.py16 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 Dbundled_inputs.py412 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 Dtest_lite_script_type.py26 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 Dtest_lite_script_module.py51 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 Dtest_bytecode.py365 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 Dtest_upgraders.py17 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.py579 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.py649 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.py1397 script_module = torch.jit._recursive.create_script_module(
1402 self.__dict__["_actual_script_module"] = script_module
H A D_script.pyi152 def _finalize_scriptmodule(script_module) -> None: ...
/aosp_15_r20/external/pytorch/test/jit/fixtures_srcs/
H A Dgenerate_models.py146 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 Dmodule_impl_check.py136 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 Dgenerate_test_torchscripts.py12 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 Dcompile_model.py72 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 Dpreprocess.py86 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 Dthroughput_benchmark.cpp43 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 Dtest_generator.py164 script_module = torch.jit.script(Foo())
166 torch.jit.save(script_module, saved_module)
H A Dtest_save_load.py413 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 Dtest_class_type.py1109 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 Dinternal.py87 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 Djit_metaprogramming_utils.py591 def script_module(*args, **kwargs): function
625 return script_module
H A Dcommon_quantization.py1198 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 Dtest_jit.py311 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()