Home
last modified time | relevance | path

Searched defs:ScriptModule (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/pytorch/torch/utils/
H A Dbundled_inputs.py41 model: torch.jit.ScriptModule,
129 model: torch.jit.ScriptModule,
183 model: torch.jit.ScriptModule,
412 def _get_bundled_inputs_attributes_and_methods(script_module: torch.jit.ScriptModule) -> Tuple[List…
H A Dmobile_optimizer.py16 script_module: torch.jit.ScriptModule,
79 def generate_mobile_module_lints(script_module: torch.jit.ScriptModule):
120 def _get_bundled_inputs_preserved_attributes(script_module: torch.jit.ScriptModule, preserved_metho…
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dverification.py502 model: torch.nn.Module | torch.jit.ScriptModule,
543 model: torch.nn.Module | torch.jit.ScriptModule,
570 model: torch.nn.Module | torch.jit.ScriptModule,
730 model: torch.nn.Module | torch.jit.ScriptModule,
1680 model: torch.nn.Module | torch.jit.ScriptModule,
H A Dutils.py1355 model: torch.nn.Module | torch.jit.ScriptModule,
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_typing.py127 def wrong_value_type(dictionary: Dict[str, torch.jit.ScriptModule]):
134 def wrong_key_type(dictionary: Dict[torch.jit.ScriptModule, str]):
142 dictionary: Dict[torch.jit.ScriptModule, torch.jit.ScriptModule]
/aosp_15_r20/external/pytorch/torch/csrc/jit/python/
H A Dmodule_python.h12 static py::handle ScriptModule = in as_module() local
/aosp_15_r20/external/pytorch/torch/jit/
H A D_freeze.py183 mod: ScriptModule, other_methods: Optional[List[str]] = None
H A D_script.py510 class ScriptModule(Module, metaclass=ScriptMeta): class
1003 class ScriptModule(torch.nn.Module): # type: ignore[no-redef] class
/aosp_15_r20/external/pytorch/test/mobile/
H A Dtest_upgraders.py17 def _save_load_mobile_module(self, script_module: torch.jit.ScriptModule):
/aosp_15_r20/external/pytorch/test/onnx/exporter/
H A Dtest_api.py105 class ScriptModule(torch.nn.Module): class
/aosp_15_r20/external/pytorch/torch/_export/
H A Dconverter.py1382 ts_model: Union[torch.jit.ScriptModule, torch.jit.ScriptFunction],
/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_pytorch_onnx_onnxruntime.py3067 class ScriptModule(torch.jit.ScriptModule): class