Home
last modified time | relevance | path

Searched refs:_load_for_lite_interpreter (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_save_load_for_op_version.py18 from torch.jit.mobile import _load_for_lite_interpreter
43 return _load_for_lite_interpreter(buffer)
95 v3_mobile_module = _load_for_lite_interpreter(
143 v3_mobile_module = _load_for_lite_interpreter(
196 v3_mobile_module = _load_for_lite_interpreter(
256 v3_mobile_module_float = _load_for_lite_interpreter(
260 v3_mobile_module_int = _load_for_lite_interpreter(
318 v3_mobile_module_float = _load_for_lite_interpreter(
322 v3_mobile_module_int = _load_for_lite_interpreter(
392 v3_mobile_module_float = _load_for_lite_interpreter(
[all …]
H A Dtest_backends.py10 from torch.jit.mobile import _load_for_lite_interpreter
588 self.mobile_module = _load_for_lite_interpreter(buffer)
689 self.mobile_module = _load_for_lite_interpreter(buffer)
780 self.mobile_module = _load_for_lite_interpreter(buffer)
/aosp_15_r20/external/pytorch/test/mobile/
H A Dtest_lite_script_module.py10 from torch.jit.mobile import _export_operator_list, _load_for_lite_interpreter
34 mobile_module = _load_for_lite_interpreter(buffer)
41 mobile_module = _load_for_lite_interpreter(fname)
56 mobile_module = _load_for_lite_interpreter(buffer)
99 mobile_module = _load_for_lite_interpreter(buffer)
133 mobile_module = _load_for_lite_interpreter(buffer)
158 mobile_module = _load_for_lite_interpreter(buffer)
169 mobile_module = _load_for_lite_interpreter(buffer)
198 mobile_module = _load_for_lite_interpreter(buffer)
320 mobile_module = _load_for_lite_interpreter(buffer)
[all …]
H A Dtest_lite_script_type.py10 from torch.jit.mobile import _load_for_lite_interpreter
35 mobile_module = _load_for_lite_interpreter(buffer) # Error here
95 mobile_module = _load_for_lite_interpreter(buffer_mobile)
118 mobile_module = _load_for_lite_interpreter(buffer_mobile)
134 mobile_module = _load_for_lite_interpreter(buffer_mobile)
161 mobile_module = _load_for_lite_interpreter(buffer_mobile)
H A Dtest_upgraders.py9 from torch.jit.mobile import _load_for_lite_interpreter
24 mobile_module = _load_for_lite_interpreter(buffer)
46 mobile_module_v2 = _load_for_lite_interpreter(str(model_path))
H A Dtest_bytecode.py19 _load_for_lite_interpreter,
308 mobile_module = _load_for_lite_interpreter(str(tmp_backport_model_path))
343 mobile_module = _load_for_lite_interpreter(bytesio)
/aosp_15_r20/external/pytorch/test/jit/fixtures_srcs/
H A Dgenerate_models.py12 from torch.jit.mobile import _export_operator_list, _load_for_lite_interpreter
149 mobile_module = _load_for_lite_interpreter(buffer)
/aosp_15_r20/external/pytorch/test/mobile/lightweight_dispatch/
H A Dtests_setup.py8 from torch.jit.mobile import _export_operator_list, _load_for_lite_interpreter
26 mobile_module = _load_for_lite_interpreter(buffer)
/aosp_15_r20/external/pytorch/torch/jit/mobile/
H A D__init__.py8 def _load_for_lite_interpreter(f, map_location=None): function
47 cpp_module = torch._C._load_for_lite_interpreter(os.fspath(f), map_location)
/aosp_15_r20/external/pytorch/test/mobile/model_test/
H A Dgen_test_model.py48 from torch.jit.mobile import _load_for_lite_interpreter
186 lite_module = _load_for_lite_interpreter(buffer)
/aosp_15_r20/external/pytorch/test/quantization/jit/
H A Dtest_ondevice_quantization.py15 from torch.jit.mobile import _load_for_lite_interpreter, LiteScriptModule
443 m = _load_for_lite_interpreter(buffer) # Error here
493 m = _load_for_lite_interpreter(buffer) # Error here
/aosp_15_r20/external/pytorch/torch/csrc/jit/operator_upgraders/
H A DREADME.md173 v3_mobile_module_float = _load_for_lite_interpreter(
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_quantization.py50 from torch.jit.mobile import _load_for_lite_interpreter
1207 mobile_module = _load_for_lite_interpreter(buffer)
/aosp_15_r20/external/pytorch/torch/_C/
H A D__init__.pyi.in453 def _load_for_lite_interpreter(