Home
last modified time | relevance | path

Searched refs:RecursiveScriptModule (Results 1 – 18 of 18) sorted by relevance

/aosp_15_r20/external/pytorch/torch/jit/
H A D_script.py595 class RecursiveScriptModule(ScriptModule): class
649 script_module = RecursiveScriptModule(cpp_module)
654 RecursiveScriptModule._finalize_scriptmodule(script_module)
868 RecursiveScriptModule, method_name
891 == _get_function_from_type(RecursiveScriptModule, "__dir__")
903 == _get_function_from_type(RecursiveScriptModule, "__bool__")
915 return RecursiveScriptModule._construct(
924 for name, item in RecursiveScriptModule.__dict__.items():
992 name not in RecursiveScriptModule.__dict__
995 setattr(RecursiveScriptModule, method.__name__, _make_fail(name))
[all …]
H A D_freeze.py11 from torch.jit._script import RecursiveScriptModule, ScriptModule
117 out = RecursiveScriptModule(torch._C._freeze_module(mod._c, preserved_attrs))
118 RecursiveScriptModule._finalize_scriptmodule(out)
H A D_recursive.py552 assert not isinstance(nn_module, torch.jit.RecursiveScriptModule)
630 script_module = torch.jit.RecursiveScriptModule._construct(cpp_module, init_fn)
727 default_attr = getattr(torch.jit.RecursiveScriptModule, attr, None)
1025 return torch.jit.RecursiveScriptModule._construct(cpp_module, init_fn)
1062 script_module = torch.jit.RecursiveScriptModule._construct(cpp_module, init_fn)
H A D_script.pyi145 class RecursiveScriptModule(ScriptModule):
220 ) -> RecursiveScriptModule: ...
H A D__init__.py41 RecursiveScriptModule,
/aosp_15_r20/external/pytorch/test/typing/pass/
H A Ddisabled_jit.py31 assert_type(scripted_module, jit.RecursiveScriptModule)
/aosp_15_r20/external/pytorch/docs/source/notes/
H A Dserialization.rst245 RecursiveScriptModule( original_name=MyModule
246 (l0): RecursiveScriptModule(original_name=Linear)
247 (l1): RecursiveScriptModule(original_name=Linear) )
/aosp_15_r20/external/pytorch/torch/utils/
H A Dmobile_optimizer.py19 backend: str = 'CPU') -> torch.jit.RecursiveScriptModule:
/aosp_15_r20/external/pytorch/torch/backends/_coreml/
H A Dpreprocess.py104 model = torch.jit.RecursiveScriptModule._construct(script_module, lambda x: None)
/aosp_15_r20/external/pytorch/test/quantization/core/experimental/
H A Dquantization_util.py95 if isinstance(model, torch.jit.RecursiveScriptModule):
/aosp_15_r20/external/pytorch/torch/nn/utils/
H A Dstateless.py280 torch.jit.RecursiveScriptModule,
/aosp_15_r20/external/pytorch/torch/ao/quantization/
H A Dobserver.py1586 if isinstance(mod, torch.jit.RecursiveScriptModule):
1602 if isinstance(module, torch.jit.RecursiveScriptModule):
1617 if isinstance(mod, torch.jit.RecursiveScriptModule):
H A Dfake_quantize.py586 if isinstance(mod, torch.jit.RecursiveScriptModule):
/aosp_15_r20/external/pytorch/torch/distributed/nn/api/
H A Dremote_module.py761 torch.jit.RecursiveScriptModule, _recursive_script_module_reducer
/aosp_15_r20/external/pytorch/torch/package/
H A Dpackage_exporter.py950 obj, torch.jit.RecursiveScriptModule
/aosp_15_r20/external/pytorch/torch/ao/ns/
H A D_numeric_suite_fx.py596 isinstance(mod, torch.jit.RecursiveScriptModule)
/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dnn_module.py780 if type(value) is torch.jit._script.RecursiveScriptModule:
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dutils.py919 torch.jit._script.RecursiveScriptModule,