Searched defs:ScriptFunction (Results 1 – 9 of 9) sorted by relevance
| /external/pytorch/torch/onnx/_internal/exporter/ |
| 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
|
| D | _compat.py | 116 | torch.export.ExportedProgram
|
| D | _core.py | 934 | torch.export.ExportedProgram
|
| /external/pytorch/torch/jit/ |
| D | _decompositions.py | 68 registry: Optional[Dict[str, torch.jit.ScriptFunction]] = None,
|
| D | _script.py | 60 ScriptFunction = torch._C.ScriptFunction variable
|
| /external/pytorch/torch/onnx/ |
| D | __init__.py | 139 | torch.export.ExportedProgram
|
| 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]
|
| /external/pytorch/torch/_export/ |
| 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],
|
| /external/pytorch/test/onnx/ |
| D | test_pytorch_onnx_no_runtime.py | 30 model: Union[torch.nn.Module, torch.jit.ScriptFunction],
|