/aosp_15_r20/external/pytorch/torch/nn/utils/ |
H A D | _named_member_accessor.py | 338 valid_keys = {name for name, _ in self.named_tensors(remove_duplicate=False)} 347 remove_duplicate: bool = True, 350 yield from self.module.named_parameters(remove_duplicate=remove_duplicate) 354 remove_duplicate: bool = True, 357 yield from self.module.named_buffers(remove_duplicate=remove_duplicate) 361 remove_duplicate: bool = True, 364 yield from self.module.named_parameters(remove_duplicate=remove_duplicate) 365 yield from self.module.named_buffers(remove_duplicate=remove_duplicate) 369 remove_duplicate: bool = True, 372 yield from self.module.named_modules(remove_duplicate=remove_duplicate)
|
H A D | stateless.py | 44 all_named_tensors.update(module.named_parameters(remove_duplicate=False)) 45 all_named_tensors.update(module.named_buffers(remove_duplicate=False))
|
/aosp_15_r20/external/pytorch/torch/nn/modules/ |
H A D | module.py | 2592 self, get_members_fn, prefix="", recurse=True, remove_duplicate: bool = True 2597 self.named_modules(prefix=prefix, remove_duplicate=remove_duplicate) 2606 if remove_duplicate: 2637 self, prefix: str = "", recurse: bool = True, remove_duplicate: bool = True 2664 remove_duplicate=remove_duplicate, 2692 self, prefix: str = "", recurse: bool = True, remove_duplicate: bool = True 2718 remove_duplicate=remove_duplicate, 2782 remove_duplicate: bool = True, 2816 if remove_duplicate: 2824 memo, submodule_prefix, remove_duplicate
|
/aosp_15_r20/external/pytorch/torch/export/ |
H A D | _trace.py | 323 for name, param in original_module.named_parameters(remove_duplicate=False): 325 for name, buffer in original_module.named_buffers(remove_duplicate=False): 330 remove_duplicate=False 337 remove_duplicate=False 499 name: type(m).__name__ for name, m in mod.named_modules(remove_duplicate=False) 778 **dict(mod.named_parameters(remove_duplicate=False)), 779 **dict(mod.named_buffers(remove_duplicate=False)), 1405 named_parameters = dict(mod.named_parameters(remove_duplicate=False)) 1406 named_buffers = dict(mod.named_buffers(remove_duplicate=False)) 1473 param_len = len(dict(mod.named_parameters(remove_duplicate=False))) [all …]
|
H A D | unflatten.py | 424 for name, _ in self.named_modules(remove_duplicate=False): 428 assert [fqn for fqn, _ in self.named_modules(remove_duplicate=False)] == list(
|
/aosp_15_r20/external/pytorch/torch/_functorch/ |
H A D | aot_autograd.py | 913 named_params = dict(mod.named_parameters(remove_duplicate=False)) 914 named_buffers = dict(mod.named_buffers(remove_duplicate=False)) 958 **dict(mod.named_parameters(remove_duplicate=False)), 959 **dict(mod.named_buffers(remove_duplicate=False)), 1158 named_parameters = dict(mod.named_parameters(remove_duplicate=False)) 1159 named_buffers = dict(mod.named_buffers(remove_duplicate=False))
|
H A D | make_functional.py | 76 all_named_members = tuple(named_members(remove_duplicate=False)) 77 unique_named_members = tuple(named_members(remove_duplicate=True))
|
/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/ |
H A D | _lower_to_native_backend.py | 661 modules = dict(model.named_modules(remove_duplicate=False)) 720 modules = dict(model.named_modules(remove_duplicate=False)) 777 named_modules = dict(model.named_modules(remove_duplicate=False)) 824 named_modules = dict(model.named_modules(remove_duplicate=False)) 851 modules = dict(model.named_modules(remove_duplicate=False)) 955 modules = dict(model.named_modules(remove_duplicate=False)) 1085 modules = dict(model.named_modules(remove_duplicate=False)) 1140 modules = dict(model.named_modules(remove_duplicate=False))
|
H A D | prepare.py | 1511 named_modules = dict(model.named_modules(remove_duplicate=False)) 1716 named_modules = dict(model.named_modules(remove_duplicate=False)) 2099 named_modules = dict(model.named_modules(remove_duplicate=False))
|
H A D | _equalize.py | 856 modules = dict(model.named_modules(remove_duplicate=False))
|
H A D | convert.py | 1072 modules = dict(model.named_modules(remove_duplicate=False))
|
/aosp_15_r20/external/pytorch/torch/distributed/nn/api/ |
H A D | remote_module.py | 411 self, prefix: str = "", recurse: bool = True, remove_duplicate: bool = True 419 self, prefix: str = "", recurse: bool = True, remove_duplicate: bool = True 436 remove_duplicate: bool = True,
|
/aosp_15_r20/external/pytorch/test/quantization/fx/ |
H A D | test_equalize_fx.py | 492 modules = dict(convert_ref.named_modules(remove_duplicate=False)) 552 modules = dict(convert_ref.named_modules(remove_duplicate=False)) 579 orig_modules = dict(orig_model.named_modules(remove_duplicate=False)) 583 eq_modules = dict(eq_model.named_modules(remove_duplicate=False))
|
/aosp_15_r20/external/fonttools/Lib/fontTools/ttLib/tables/ |
H A D | otBase.py | 207 return writer.getAllData(remove_duplicate=False) 676 def getAllData(self, remove_duplicate=True): argument 678 if remove_duplicate:
|
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/ |
H A D | python.h | 144 …duleType& module, py::object /* unused */, std::string prefix, bool remove_duplicate /* unused */)… in add_module_bindings()
|
/aosp_15_r20/external/pytorch/torch/_export/ |
H A D | utils.py | 119 recurse=True, remove_duplicate=False 124 recurse=True, remove_duplicate=False
|
/aosp_15_r20/external/pytorch/torch/ao/ns/ |
H A D | _numeric_suite_fx.py | 908 modules = dict(mt.named_modules(remove_duplicate=False)) 1010 modules = dict(mt.named_modules(remove_duplicate=False))
|
/aosp_15_r20/external/pytorch/torch/ao/quantization/pt2e/ |
H A D | prepare.py | 489 named_modules = dict(model.named_modules(remove_duplicate=False))
|
/aosp_15_r20/external/pytorch/test/export/ |
H A D | test_unflatten.py | 753 fqn_list = [x for x, _ in unflattened.named_modules(remove_duplicate=False)] 756 [x for x, _ in mod.named_modules(remove_duplicate=False)],
|
/aosp_15_r20/external/pytorch/torch/_dynamo/ |
H A D | eval_frame.py | 1361 named_parameters = dict(graph.named_parameters(remove_duplicate=False)) 1362 named_buffers = dict(graph.named_buffers(remove_duplicate=False))
|
/aosp_15_r20/external/pytorch/torch/_inductor/ |
H A D | compile_fx.py | 160 for name, param in mod.named_parameters(remove_duplicate=False): 168 for name, buffer in mod.named_buffers(remove_duplicate=False):
|
/aosp_15_r20/external/pytorch/torch/distributed/_composable/fsdp/ |
H A D | _fsdp_param_group.py | 601 for _, submodule in module.named_modules(remove_duplicate=False):
|
/aosp_15_r20/external/pytorch/test/dynamo/ |
H A D | test_aot_autograd.py | 814 named_parameters = dict(g_mod.named_parameters(remove_duplicate=False)) 815 named_buffers = dict(g_mod.named_buffers(remove_duplicate=False))
|
/aosp_15_r20/external/pytorch/torch/testing/_internal/ |
H A D | common_quantization.py | 798 modules = dict(graph_module.named_modules(remove_duplicate=False)) 850 modules = dict(graph_module.named_modules(remove_duplicate=False))
|
/aosp_15_r20/external/pytorch/torch/distributed/fsdp/ |
H A D | _flat_param.py | 651 for submodule_name, submodule in module.named_modules(remove_duplicate=False):
|