Home
last modified time | relevance | path

Searched refs:ScriptModule (Results 1 – 25 of 79) sorted by relevance

1234

/aosp_15_r20/external/pytorch/torch/utils/
H A Dbundled_inputs.py41 model: torch.jit.ScriptModule, argument
46 ) -> torch.jit.ScriptModule:
107 if not isinstance(model, torch.jit.ScriptModule):
129 model: torch.jit.ScriptModule, argument
165 if not isinstance(model, torch.jit.ScriptModule):
183 model: torch.jit.ScriptModule, argument
238 if not isinstance(model, torch.jit.ScriptModule):
412 def _get_bundled_inputs_attributes_and_methods(script_module: torch.jit.ScriptModule) -> Tuple[List… argument
H A Dmobile_optimizer.py16 script_module: torch.jit.ScriptModule, argument
33 if not isinstance(script_module, torch.jit.ScriptModule):
79 def generate_mobile_module_lints(script_module: torch.jit.ScriptModule): argument
89 if not isinstance(script_module, torch.jit.ScriptModule):
120 def _get_bundled_inputs_preserved_attributes(script_module: torch.jit.ScriptModule, preserved_metho… argument
H A Dmkldnn.py5 class MkldnnLinear(torch.jit.ScriptModule):
37 class _MkldnnConvNd(torch.jit.ScriptModule):
134 class MkldnnBatchNorm(torch.jit.ScriptModule):
185 class MkldnnPrelu(torch.jit.ScriptModule):
/aosp_15_r20/external/pytorch/torch/jit/
H A D_freeze.py11 from torch.jit._script import RecursiveScriptModule, ScriptModule
103 if not isinstance(mod, ScriptModule):
183 mod: ScriptModule, other_methods: Optional[List[str]] = None argument
184 ) -> ScriptModule:
214 if not isinstance(mod, ScriptModule):
H A Dquantized.py5 class QuantizedLinear(torch.jit.ScriptModule):
14 class QuantizedLinearFP16(torch.jit.ScriptModule):
24 class QuantizedRNNCellBase(torch.jit.ScriptModule):
58 class QuantizedRNNBase(torch.jit.ScriptModule):
H A D_script.py256 if isinstance(v, ScriptModule):
510 class ScriptModule(Module, metaclass=ScriptMeta): class
595 class RecursiveScriptModule(ScriptModule):
927 if name.startswith("__") or hasattr(ScriptModule, name):
932 setattr(ScriptModule, name, item)
1003 class ScriptModule(torch.nn.Module): # type: ignore[no-redef] class
1007 class RecursiveScriptModule(ScriptModule): # type: ignore[no-redef]
1036 sub_module, ScriptModule
1107 if isinstance(obj, ScriptModule):
H A D_trace.py31 from torch.jit._script import _CachedForward, script, ScriptModule
621 if isinstance(mod, ScriptModule):
684 if isinstance(func, torch.jit.ScriptModule):
1093 if not isinstance(traced_func_for_export, torch.jit.ScriptModule):
1334 class TracedModule(ScriptModule):
/aosp_15_r20/external/pytorch/benchmarks/fastrnns/
H A Dcustom_lstms.py122 class LSTMCell(jit.ScriptModule):
156 class LayerNorm(jit.ScriptModule):
182 class LayerNormLSTMCell(jit.ScriptModule):
221 class LSTMLayer(jit.ScriptModule):
238 class ReverseLSTMLayer(jit.ScriptModule):
255 class BidirLSTMLayer(jit.ScriptModule):
292 class StackedLSTM(jit.ScriptModule):
322 class StackedLSTM2(jit.ScriptModule):
349 class StackedLSTMWithDropout(jit.ScriptModule):
/aosp_15_r20/external/pytorch/torch/_C/
H A D__init__.pyi.in300 module: ScriptModule,
304 ) -> ScriptModule: ...
307 module: torch.jit.ScriptModule,
317 def _jit_pass_remove_dropout(module: torch.jit.ScriptModule): ...
329 module: torch.jit.ScriptModule,
332 ) -> torch.jit.ScriptModule: ...
334 module: torch.jit.ScriptModule,
337 ) -> torch.jit.ScriptModule: ...
339 module: torch.jit.ScriptModule,
342 ) -> torch.jit.ScriptModule: ...
[all …]
/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_pytorch_onnx_no_runtime.py94 class Foo(torch.jit.ScriptModule):
121 class ModuleToExport(torch.jit.ScriptModule):
146 class PythonModule(torch.jit.ScriptModule):
151 class ModuleToExport(torch.jit.ScriptModule):
171 class ModuleToExport(torch.jit.ScriptModule):
185 class ModuleToInline(torch.jit.ScriptModule):
190 class ModuleToExport(torch.jit.ScriptModule):
204 class ModuleToExport(torch.jit.ScriptModule):
219 class ModuleToExport(torch.jit.ScriptModule):
232 class ModuleToExport(torch.jit.ScriptModule):
[all …]
H A Dtest_pytorch_onnx_onnxruntime.py462 class Data(torch.jit.ScriptModule):
934 class MyModel(torch.jit.ScriptModule):
964 class ScalarInputModel(torch.jit.ScriptModule):
975 class TensorInputModel(torch.jit.ScriptModule):
986 class FloatInputModel(torch.jit.ScriptModule):
994 class InputWithDtypeModel(torch.jit.ScriptModule):
1005 class MixedInputModel(torch.jit.ScriptModule):
1019 class MyModel(torch.jit.ScriptModule):
1040 class MyModel(torch.jit.ScriptModule):
1126 class ClampMaxModel(torch.jit.ScriptModule):
[all …]
/aosp_15_r20/external/pytorch/torch/_C_flatbuffer/
H A D__init__.pyi2 from torch._C import LiteScriptModule, ScriptModule
9 def _save_jit_module(m: ScriptModule, filename: str): ...
11 def _save_jit_module_to_bytes(m: ScriptModule) -> bytes: ...
/aosp_15_r20/external/pytorch/test/
H A Dtest_jit.py342 self.bar = torch.jit.ScriptModule()
412 m = torch.jit.ScriptModule()
461 class M(torch.jit.ScriptModule):
479 class MyModule(torch.jit.ScriptModule):
533 class Foo(torch.jit.ScriptModule):
990 class MyModule(torch.jit.ScriptModule):
1011 class MyModule(torch.jit.ScriptModule):
2452 class Sub(torch.jit.ScriptModule):
2461 class M(torch.jit.ScriptModule):
2652 class Foo(torch.jit.ScriptModule):
[all …]
/aosp_15_r20/external/pytorch/docs/source/
H A Djit.rst60 ScriptModule
262 TorchScript provides a code pretty-printer for all :class:`ScriptModule` instances. This
286 A :class:`ScriptModule` with a single ``forward`` method will have an attribute
287 ``code``, which you can use to inspect the :class:`ScriptModule`'s code.
288 If the :class:`ScriptModule` has more than one method, you will need to access
290 code of a method named ``foo`` on a :class:`ScriptModule` by accessing ``.foo.code``.
385 by a :class:`ScriptModule` is correct, in both automated and manual fashion, as
587 Q: How do I store attributes on a :class:`ScriptModule`?
609 compiler of attributes on :class:`ScriptModule`:
661 :class:`ScriptModule`\s, instead of inheriting from ``torch.jit.ScriptModule``.
[all …]
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_logging.py25 class ModuleThatLogs(torch.jit.ScriptModule):
66 class ModuleThatTimes(torch.jit.ScriptModule):
85 class ModuleThatTimes(torch.jit.ScriptModule):
H A Dtest_type_sharing.py27 if not isinstance(m1, torch.jit.ScriptModule):
29 if not isinstance(m2, torch.jit.ScriptModule):
34 if not isinstance(m1, torch.jit.ScriptModule):
36 if not isinstance(m2, torch.jit.ScriptModule):
524 class A(torch.jit.ScriptModule):
H A Dtest_typing.py127 def wrong_value_type(dictionary: Dict[str, torch.jit.ScriptModule]): argument
134 def wrong_key_type(dictionary: Dict[torch.jit.ScriptModule, str]): argument
142 dictionary: Dict[torch.jit.ScriptModule, torch.jit.ScriptModule] argument
/aosp_15_r20/external/pytorch/torch/nn/parallel/
H A Dreplicate.py22 from torch.jit import ScriptModule
32 return isinstance(module, torch.jit.ScriptModule)
44 return torch.jit.ScriptModule()
/aosp_15_r20/external/pytorch/test/mobile/
H A Dtest_lite_script_module.py332 class FooTest(torch.jit.ScriptModule):
347 class FooTest2(torch.jit.ScriptModule):
367 class FooTest3(torch.jit.ScriptModule):
395 class FooTest4(torch.jit.ScriptModule):
410 class FooTest5(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
14 if (py::isinstance(obj, ScriptModule)) { in as_module()
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dutils.py176 model: torch.nn.Module | torch.jit.ScriptModule | torch.jit.ScriptFunction,
962 ) -> tuple[_C.Graph, list[_C.IValue], Any | None, _C.ScriptModule | None]:
963 if isinstance(model, (torch.jit.ScriptFunction, torch.jit.ScriptModule)):
968 if isinstance(model, torch.jit.ScriptModule):
975 cast(_C.ScriptModule, model._c), preserveParameters=True
1131 is_script = isinstance(model, (torch.jit.ScriptFunction, torch.jit.ScriptModule))
1355 model: torch.nn.Module | torch.jit.ScriptModule, argument
1536 if isinstance(model, (torch.nn.Module, torch.jit.ScriptModule)):
H A Dverification.py37 _ModelType = Union[torch.nn.Module, torch.jit.ScriptModule]
502 model: torch.nn.Module | torch.jit.ScriptModule, argument
543 model: torch.nn.Module | torch.jit.ScriptModule, argument
570 model: torch.nn.Module | torch.jit.ScriptModule, argument
730 model: torch.nn.Module | torch.jit.ScriptModule, argument
1680 model: torch.nn.Module | torch.jit.ScriptModule, argument
/aosp_15_r20/external/pytorch/torch/distributed/rpc/
H A Drref_proxy.py26 bypass_type = issubclass(rref_type, torch.jit.ScriptModule) or issubclass(
27 rref_type, torch._C.ScriptModule
/aosp_15_r20/external/pytorch/test/onnx/exporter/
H A Dtest_api.py105 class ScriptModule(torch.nn.Module): class
109 self.assert_export(torch.jit.script(ScriptModule()), (torch.randn(1, 1, 2),))
/aosp_15_r20/external/pytorch/torch/csrc/jit/docs/
H A Dserialization.md112 ("code object") and outputs Python source code. `ScriptModule`s are
221 A model is really a top-level `ScriptModule` with any number of submodules,
367 TorchScript class, or a `ScriptModule`. Owns other its attribute types
407 1. For `ScriptModule`s, since every `ScriptModule` is a singleton class in
408 the JIT, a user that is constructing multiple `ScriptModule`s will create

1234