Home
last modified time | relevance | path

Searched refs:ModuleDict (Results 1 – 25 of 36) sorted by relevance

12

/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dmoduledict.cpp24 ModuleDict dict(list); in TEST_F()
39 ModuleDict dict(ordereddict); in TEST_F()
49 ModuleDict dict; in TEST_F()
66 ModuleDict updatedict(list2); in TEST_F()
88 ModuleDict dict(list1); in TEST_F()
108 ModuleDict dict2(list3); in TEST_F()
126 ModuleDict dict(ordereddict); in TEST_F()
147 ModuleDict dict(ordereddict); in TEST_F()
168 ModuleDict dict(ordereddict); in TEST_F()
177 ModuleDict first(ordereddict); in TEST_F()
[all …]
/aosp_15_r20/external/pytorch/torch/ao/pruning/_experimental/pruner/
H A Dprune_functions.py34 parametrization_dict = cast(nn.ModuleDict, next_layer.parametrizations)
115 parametrization_dict = cast(nn.ModuleDict, linear.parametrizations)
157 parametrization_dict = cast(nn.ModuleDict, linear2.parametrizations)
173 parametrization_dict = cast(nn.ModuleDict, conv2d.parametrizations)
189 parametrization_dict = cast(nn.ModuleDict, conv2d_1.parametrizations)
239 parametrization_dict = cast(nn.ModuleDict, conv2d_1.parametrizations)
275 nn.ModuleDict, conv2d_2.parametrizations
320 parametrization_dict = cast(nn.ModuleDict, linear.parametrizations)
351 parametrization_dict = cast(nn.ModuleDict, linear.parametrizations)
378 parametrization_dict = cast(nn.ModuleDict, lstm.parametrizations)
[all …]
H A Dmatch_utils.py14 modules: Dict[str, nn.ModuleDict], argument
39 modules: Dict[str, nn.ModuleDict], argument
/aosp_15_r20/external/pytorch/torch/nn/utils/
H A Dparametrize.py12 from torch.nn.modules.container import Module, ModuleDict, ModuleList
597 assert isinstance(module.parametrizations, ModuleDict) # Make mypy happy
617 module.parametrizations = ModuleDict()
621 assert isinstance(module.parametrizations, ModuleDict) # Make mypy happy
644 if parametrizations is None or not isinstance(parametrizations, ModuleDict):
686 assert isinstance(module.parametrizations, ModuleDict) # Make mypy happy
778 assert isinstance(from_module.parametrizations, ModuleDict) # for mypy
781 parameters_to_transfer: Union[list, ModuleDict] = (
798 assert isinstance(to_module.parametrizations, ModuleDict) # for mypy
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_module_containers.py68 self.moduledict = nn.ModuleDict(modules)
158 class CustomModuleDict(nn.ModuleDict):
273 class CustomModuleDict(CustomModuleInterface, torch.nn.ModuleDict):
276 torch.nn.ModuleDict.__init__(self, modules)
366 self.moduledict = torch.nn.ModuleDict(
382 self.moduledict = torch.nn.ModuleDict({"foo": None, "bar": None})
396 self.moduledict = torch.nn.ModuleDict({"foo": None, "bar": None})
435 class CustomModuleDict(CustomModuleInterface, torch.nn.ModuleDict):
438 torch.nn.ModuleDict.__init__(self, modules)
482 self.d = torch.nn.ModuleDict({"module": ImplementsInterface()})
[all …]
H A Dtest_type_sharing.py506 self.dict = torch.nn.ModuleDict(s)
H A Dtest_recursive_script.py309 nn.ModuleDict(OrderedDict([("conv", conv), ("linear", linear)]))
/aosp_15_r20/external/pytorch/test/distributed/_shard/
H A Dtest_sharder.py34 self.embedding_bags: nn.ModuleDict = nn.ModuleDict()
56 self.embedding_bags: nn.ModuleDict = nn.ModuleDict()
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_modules.py365 class ModuleDict(torch.nn.Module): class
368 self.layers = torch.nn.ModuleDict(
414 self.layers = torch.nn.ModuleDict(
530 class _DenseBlock(torch.nn.ModuleDict):
819 class EnumValues(torch.nn.ModuleDict):
836 class AccessByKeys(torch.nn.ModuleDict):
992 self.layer5 = torch.nn.ModuleDict(
1029 class ModuleGuardNameIsValid(torch.nn.ModuleDict):
1189 test_moduledict = make_test(ModuleDict())
1404 module_dict = torch.nn.ModuleDict({"foo": torch.nn.Conv2d(1, 1, 1)})
[all …]
/aosp_15_r20/external/pytorch/torch/jit/
H A D_recursive.py193 if isinstance(nn_module, (torch.nn.ModuleDict)):
653 nn_module, (torch.nn.ModuleList, torch.nn.Sequential, torch.nn.ModuleDict)
659 isinstance(nn_module, torch.nn.ModuleDict)
1014 for name, cpp_module in torch._C.ModuleDict(script_module._c).items():
H A D_script.py230 super().__init__(torch._C.ModuleDict(module))
686 for name, cpp_module in torch._C.ModuleDict(self._c).items():
/aosp_15_r20/external/pytorch/docs/source/
H A Djit_language_reference_v2.rst338 "torch.nn.ModuleList" | "torch.nn.ModuleDict" | ...
343 Special Note on torch.nn.ModuleList and torch.nn.ModuleDict
346 Although ``torch.nn.ModuleList`` and ``torch.nn.ModuleDict`` are defined as a list and dictionary i…
349 * In TorchScript, instances of ``torch.nn.ModuleList`` or ``torch.nn.ModuleDict`` are immutable.
350 …ist`` or ``torch.nn.ModuleDict`` is completely unrolled so that elements of ``torch.nn.ModuleList`…
849 "torch.nn.ModuleDict" | ...
994 * If the primary is a ``ModuleDict``, the expression must be a ``stringliteral``.
1478 …`, ``tuples``, ``dictionaries``, ``strings``, ``torch.nn.ModuleList``, and ``torch.nn.ModuleDict``.
1557 …`, ``tuples``, ``dictionaries``, ``strings``, ``torch.nn.ModuleList``, and ``torch.nn.ModuleDict``.
1567 …``, ``tuples``, ``dictionaries``, ``strings``, ``torch.nn.ModuleList`` and ``torch.nn.ModuleDict``.
H A Ddistributed.pipelining.rst152 # Using a ModuleDict lets us delete layers without affecting names,
154 self.layers = torch.nn.ModuleDict()
H A Dnn.rst41 ModuleDict
/aosp_15_r20/external/pytorch/torch/nn/modules/
H A D__init__.py47 ModuleDict,
H A Dcontainer.py449 class ModuleDict(Module): class
568 if isinstance(modules, (OrderedDict, ModuleDict, container_abcs.Mapping)):
/aosp_15_r20/external/pytorch/test/distributed/_composable/fsdp/
H A Dtest_fully_shard_state.py75 model = nn.ModuleDict({"1": MLP(8), "2": MLP(8)})
/aosp_15_r20/external/pytorch/test/
H A Dtest_module_tracker.py32 self.b = nn.ModuleDict({"nest": Foo()})
/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dnn_module.py161 if isinstance(base, torch.nn.ModuleDict):
645 and isinstance(module, (torch.nn.ModuleDict, torch.nn.ParameterDict))
655 torch.nn.ModuleDict.__getitem__,
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/modules/container/
H A Dmoduledict.h259 TORCH_MODULE(ModuleDict);
/aosp_15_r20/external/pytorch/benchmarks/dynamo/pr_time_benchmarks/benchmarks/
H A Dbasic_modules_benchmarks.py47 self.layer5 = torch.nn.ModuleDict(
/aosp_15_r20/external/pytorch/test/distributed/_tools/
H A Dtest_mod_tracker.py31 self.b = torch.nn.ModuleDict({"nest": Foo()})
/aosp_15_r20/external/pytorch/torch/distributed/_composable/fsdp/
H A Dfully_shard.py105 if isinstance(module, (nn.ModuleList, nn.ModuleDict)):
/aosp_15_r20/external/pytorch/torch/distributed/fsdp/
H A Dwrap.py405 size_based_auto_wrap_policy.EXCLUDE_WRAP_MODULES = {nn.ModuleList, nn.ModuleDict} # type: ignore[a…
/aosp_15_r20/external/pytorch/test/cpp_api_parity/
H A Dparity-tracker.md8 torch::nn::ModuleDict|No|No

12