Home
last modified time | relevance | path

Searched refs:remove_duplicate (Results 1 – 25 of 30) sorted by relevance

12

/aosp_15_r20/external/pytorch/torch/nn/utils/
H A D_named_member_accessor.py338 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 Dstateless.py44 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 Dmodule.py2592 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.py323 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 Dunflatten.py424 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 Daot_autograd.py913 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 Dmake_functional.py76 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.py661 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 Dprepare.py1511 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.py856 modules = dict(model.named_modules(remove_duplicate=False))
H A Dconvert.py1072 modules = dict(model.named_modules(remove_duplicate=False))
/aosp_15_r20/external/pytorch/torch/distributed/nn/api/
H A Dremote_module.py411 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 Dtest_equalize_fx.py492 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 DotBase.py207 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 Dpython.h144 …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 Dutils.py119 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.py908 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 Dprepare.py489 named_modules = dict(model.named_modules(remove_duplicate=False))
/aosp_15_r20/external/pytorch/test/export/
H A Dtest_unflatten.py753 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 Deval_frame.py1361 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 Dcompile_fx.py160 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.py601 for _, submodule in module.named_modules(remove_duplicate=False):
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_aot_autograd.py814 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 Dcommon_quantization.py798 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.py651 for submodule_name, submodule in module.named_modules(remove_duplicate=False):

12