Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/distributed/checkpoint/
H A Dstate_dict.py153 fqn_param_mapping: Dict[ variable in _StateDictInfo
289 fqn_param_mapping: Dict[
300 fqn = fqn_param_mapping.get(param, None)
302 cast(Set[str], fqn_param_mapping[param]).update(fqns)
303 shared_params_mapping[param] = fqn_param_mapping[param]
306 fqn_param_mapping[param] = fqns.copy()
309 fqn_param_mapping[fqn] = param
384 fqn_param_mapping=fqn_param_mapping,
704 for fqn in info.fqn_param_mapping[param]:
831 for fqn in info.fqn_param_mapping[param]: