Searched defs:ScriptFunction (Results 1 – 11 of 11) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/onnx/_internal/exporter/ |
| H A D | _capture_strategies.py | 81 model: torch.nn.Module | torch.jit.ScriptFunction,
107 def _enter(self, model: torch.nn.Module | torch.jit.ScriptFunction) -> None:
110 def _success(self, model: torch.nn.Module | torch.jit.ScriptFunction) -> None:
114 self, model: torch.nn.Module | torch.jit.ScriptFunction, e: Exception
|
| H A D | _compat.py | 116 | torch.export.ExportedProgram
|
| H A D | _core.py | 934 | torch.export.ExportedProgram
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | utils.py | 110 def disable_apex_o2_state_dict_hook(model: torch.nn.Module | torch.jit.ScriptFunction):
176 model: torch.nn.Module | torch.jit.ScriptModule | torch.jit.ScriptFunction,
961 model: torch.nn.Module | torch.jit.ScriptFunction, args: Sequence[Any]
|
| H A D | __init__.py | 139 | torch.export.ExportedProgram
|
| /aosp_15_r20/external/pytorch/torch/jit/ |
| H A D | _decompositions.py | 68 registry: Optional[Dict[str, torch.jit.ScriptFunction]] = None,
|
| H A D | _script.py | 60 ScriptFunction = torch._C.ScriptFunction variable
|
| /aosp_15_r20/external/pytorch/torch/_export/ |
| H A D | converter.py | 73 model: Union[torch.nn.Module, torch.jit.ScriptFunction], args: Sequence[Any]
1382 ts_model: Union[torch.jit.ScriptModule, torch.jit.ScriptFunction],
|
| /aosp_15_r20/external/pytorch/test/onnx/ |
| H A D | test_pytorch_onnx_no_runtime.py | 30 model: Union[torch.nn.Module, torch.jit.ScriptFunction],
|
| /aosp_15_r20/prebuilts/jdk/jdk8/linux-x86/jre/lib/ext/ |
| D | nashorn.jar | META-INF/
META-INF/MANIFEST.MF
META-INF/services/javax. ... |
| /aosp_15_r20/prebuilts/jdk/jdk8/darwin-x86/jre/lib/ext/ |
| D | nashorn.jar | META-INF/
META-INF/MANIFEST.MF
META-INF/services/javax. ... |