| /aosp_15_r20/external/pytorch/torch/distributed/rpc/ |
| H A D | api.py | 666 if isinstance(wrapped, torch.jit.ScriptFunction):
673 elif isinstance(func, torch.jit.ScriptFunction):
722 if isinstance(wrapped, torch.jit.ScriptFunction):
729 elif isinstance(func, torch.jit.ScriptFunction):
949 if isinstance(func, torch.jit.ScriptFunction)
|
| /aosp_15_r20/external/pytorch/torch/onnx/_internal/exporter/ |
| H A D | _capture_strategies.py | 81 model: torch.nn.Module | torch.jit.ScriptFunction, argument
107 def _enter(self, model: torch.nn.Module | torch.jit.ScriptFunction) -> None: argument
110 def _success(self, model: torch.nn.Module | torch.jit.ScriptFunction) -> None: argument
114 self, model: torch.nn.Module | torch.jit.ScriptFunction, e: Exception argument
|
| H A D | _compat.py | 118 | torch.jit.ScriptFunction,
|
| H A D | _core.py | 937 | torch.jit.ScriptFunction,
|
| /aosp_15_r20/external/pytorch/test/typing/pass/ |
| H A D | disabled_jit.py | 7 from torch import jit, nn, ScriptDict, ScriptFunction, ScriptList
36 relu: ScriptFunction = jit.script(nn.functional.relu)
|
| /aosp_15_r20/external/pytorch/torch/jit/ |
| H A D | _script.py | 59 torch._C.ScriptFunction.graph_for = _graph_for # type: ignore[attr-defined]
60 ScriptFunction = torch._C.ScriptFunction variable
61 ScriptFunction.__doc__ = """
65 set_module(ScriptFunction, "torch.jit")
74 ScriptFunction.__reduce__ = _reduce # type: ignore[assignment]
1109 if isinstance(obj, ScriptFunction):
|
| H A D | _decompositions.py | 15 decomposition_table: Dict[str, torch.jit.ScriptFunction] = {}
68 registry: Optional[Dict[str, torch.jit.ScriptFunction]] = None, argument
|
| H A D | _script.pyi | 52 ScriptFunction = torch._C.ScriptFunction
236 ) -> ScriptFunction: ...
|
| H A D | _state.py | 126 assert isinstance(value, torch.jit.ScriptFunction)
|
| H A D | __init__.py | 44 ScriptFunction,
|
| H A D | _recursive.py | 388 if isinstance(value, torch.jit.ScriptFunction):
|
| /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): argument
115 if not isinstance(model, torch.jit.ScriptFunction):
176 model: torch.nn.Module | torch.jit.ScriptModule | torch.jit.ScriptFunction, argument
961 model: torch.nn.Module | torch.jit.ScriptFunction, args: Sequence[Any] argument
963 if isinstance(model, (torch.jit.ScriptFunction, torch.jit.ScriptModule)):
1131 is_script = isinstance(model, (torch.jit.ScriptFunction, torch.jit.ScriptModule))
|
| H A D | __init__.py | 141 | torch.jit.ScriptFunction,
|
| /aosp_15_r20/external/pytorch/torch/_export/ |
| H A D | converter.py | 73 model: Union[torch.nn.Module, torch.jit.ScriptFunction], args: Sequence[Any] argument
75 if isinstance(model, (torch.jit.ScriptFunction, torch.jit.ScriptModule)):
1382 ts_model: Union[torch.jit.ScriptModule, torch.jit.ScriptFunction], argument
1396 if not isinstance(self.ts_model, torch._C.ScriptFunction)
1399 if not isinstance(self.ts_model, torch._C.ScriptFunction):
1455 if not isinstance(self.ts_model, torch._C.ScriptFunction):
|
| /aosp_15_r20/external/pytorch/torch/nn/utils/ |
| H A D | stateless.py | 282 torch.jit.ScriptFunction,
|
| /aosp_15_r20/external/pytorch/benchmarks/instruction_counts/core/ |
| H A D | expand.py | 82 jit_model, (torch.jit.ScriptFunction, torch.jit.ScriptModule)
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _jit_internal.py | 137 if isinstance(obj, torch._C.ScriptFunction):
1534 elif isinstance(obj, torch.jit.ScriptFunction):
|
| /aosp_15_r20/external/pytorch/torch/utils/tensorboard/ |
| H A D | _pytorch_graph.py | 363 if not isinstance(model, torch.jit.ScriptFunction):
|
| /aosp_15_r20/external/pytorch/torch/_C/ |
| H A D | __init__.pyi.in | 443 def _create_function_from_graph(qualname: str, graph: Graph) -> ScriptFunction: ...
1000 def find_function(self, name: str) -> ScriptFunction: ...
1001 def __getattr__(self, name: str) -> ScriptFunction: ...
1009 def get_functions(self) -> List[ScriptFunction]: ...
1015 ) -> ScriptFunction: ...
1032 class ScriptFunction(Generic[P, ReturnVal]):
|
| /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. ... |
| /aosp_15_r20/external/pytorch/test/onnx/ |
| H A D | onnx_test_common.py | 192 model, (torch.jit.ScriptModule, torch.jit.ScriptFunction)
|
| /aosp_15_r20/external/pytorch/torch/utils/benchmark/utils/valgrind_wrapper/ |
| H A D | timer_interface.py | 302 Serialization.TORCH_JIT: (torch.jit.ScriptFunction, torch.jit.ScriptModule),
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/variables/ |
| H A D | functions.py | 135 fn, (types.FunctionType, torch.jit.ScriptFunction)
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | utils.py | 576 torch.jit.ScriptFunction,
920 torch.jit.ScriptFunction,
|