Home
last modified time | relevance | path

Searched refs:_named_parameters_with_duplicates (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/fsdp/
H A D_common_utils.py264 def _named_parameters_with_duplicates( function
317 for param_name, param in _named_parameters_with_duplicates(
361 [key for key, _ in _named_parameters_with_duplicates(model)],
H A D_optim_utils.py37 _named_parameters_with_duplicates,
1048 for param_name, param in _named_parameters_with_duplicates(
1064 [fqn for fqn, _ in _named_parameters_with_duplicates(model)],
1088 for key, _ in _named_parameters_with_duplicates(model):
2069 [fqn for fqn, _ in _named_parameters_with_duplicates(model)],
H A D_init_utils.py37 _named_parameters_with_duplicates,
1181 for param_name, param in _named_parameters_with_duplicates(fsdp_module):
H A D_flat_param.py33 _named_parameters_with_duplicates,
652 for param_name, param in _named_parameters_with_duplicates(
/aosp_15_r20/external/pytorch/torch/distributed/_composable/fsdp/
H A D_fsdp_param_group.py10 from torch.distributed.fsdp._common_utils import _named_parameters_with_duplicates
602 for param_name, param in _named_parameters_with_duplicates(