Searched refs:LiteScriptModule (Results 1 – 4 of 4) sorted by relevance
2 from torch._C import LiteScriptModule, ScriptModule8 def _save_mobile_module(m: LiteScriptModule, filename: str): ...10 def _save_mobile_module_to_bytes(m: LiteScriptModule) -> bytes: ...
53 return LiteScriptModule(cpp_module)56 class LiteScriptModule: class74 def _export_operator_list(module: LiteScriptModule): argument
15 from torch.jit.mobile import _load_for_lite_interpreter, LiteScriptModule455 m = LiteScriptModule(torch._C._load_mobile_module_from_bytes(bytes))
461 def _export_operator_list(module: LiteScriptModule): ...462 def _quantize_ondevice_ptq_dynamic(module: LiteScriptModule, method_name: str): ...1025 class LiteScriptModule:2323 def _save_mobile_module(m: LiteScriptModule, filename: str): ...2325 def _save_mobile_module_to_bytes(m: LiteScriptModule) -> bytes: ...