| /external/pytorch/test/cpp/api/ |
| D | moduledict.cpp | 24 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 …]
|
| /external/pytorch/torch/ao/pruning/_experimental/pruner/ |
| D | prune_functions.py | 34 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 …]
|
| D | match_utils.py | 14 modules: Dict[str, nn.ModuleDict], argument
39 modules: Dict[str, nn.ModuleDict], argument
|
| /external/pytorch/torch/nn/utils/ |
| D | parametrize.py | 12 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
|
| /external/pytorch/test/jit/ |
| D | test_module_containers.py | 68 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 …]
|
| /external/pytorch/test/distributed/_shard/ |
| D | test_sharder.py | 34 self.embedding_bags: nn.ModuleDict = nn.ModuleDict()
56 self.embedding_bags: nn.ModuleDict = nn.ModuleDict()
|
| /external/python/cpython3/Tools/clinic/libclinic/ |
| D | app.py | 18 Module, Function, ClassDict, ModuleDict)
101 self.modules: ModuleDict = {}
|
| D | function.py | 19 ModuleDict = dict[str, "Module"] variable
30 self.modules: ModuleDict = {}
|
| /external/pytorch/torch/nn/modules/ |
| D | __init__.py | 47 ModuleDict,
|
| D | container.py | 449 class ModuleDict(Module): class
568 if isinstance(modules, (OrderedDict, ModuleDict, container_abcs.Mapping)):
|
| /external/pytorch/test/distributed/_composable/fsdp/ |
| D | test_fully_shard_state.py | 75 model = nn.ModuleDict({"1": MLP(8), "2": MLP(8)})
|
| /external/pytorch/test/ |
| D | test_module_tracker.py | 32 self.b = nn.ModuleDict({"nest": Foo()})
|
| /external/pytorch/torch/jit/ |
| D | _recursive.py | 193 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():
|
| D | _script.py | 230 super().__init__(torch._C.ModuleDict(module))
686 for name, cpp_module in torch._C.ModuleDict(self._c).items():
|
| /external/pytorch/test/dynamo/ |
| D | test_modules.py | 365 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 …]
|
| /external/pytorch/torch/csrc/api/include/torch/nn/modules/container/ |
| D | moduledict.h | 259 TORCH_MODULE(ModuleDict);
|
| /external/pytorch/benchmarks/dynamo/pr_time_benchmarks/benchmarks/ |
| D | basic_modules_benchmarks.py | 47 self.layer5 = torch.nn.ModuleDict(
|
| /external/pytorch/test/distributed/_tools/ |
| D | test_mod_tracker.py | 31 self.b = torch.nn.ModuleDict({"nest": Foo()})
|
| /external/pytorch/docs/source/ |
| D | jit_language_reference_v2.rst | 338 "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``.
|
| D | distributed.pipelining.rst | 152 # Using a ModuleDict lets us delete layers without affecting names,
154 self.layers = torch.nn.ModuleDict()
|
| D | nn.rst | 41 ModuleDict
|
| /external/pytorch/torch/distributed/_composable/fsdp/ |
| D | fully_shard.py | 105 if isinstance(module, (nn.ModuleList, nn.ModuleDict)):
|
| /external/pytorch/torch/_dynamo/variables/ |
| D | nn_module.py | 161 if isinstance(base, torch.nn.ModuleDict):
645 and isinstance(module, (torch.nn.ModuleDict, torch.nn.ParameterDict))
655 torch.nn.ModuleDict.__getitem__,
|
| /external/pytorch/torch/distributed/fsdp/ |
| D | wrap.py | 405 size_based_auto_wrap_policy.EXCLUDE_WRAP_MODULES = {nn.ModuleList, nn.ModuleDict} # type: ignore[a…
|
| /external/pytorch/test/cpp_api_parity/ |
| D | parity-tracker.md | 8 torch::nn::ModuleDict|No|No
|