Searched refs:shared_params_mapping (Results 1 – 1 of 1) sorted by relevance
156 shared_params_mapping: Dict[ variable in _StateDictInfo292 shared_params_mapping: Dict[303 shared_params_mapping[param] = fqn_param_mapping[param]311 for param_, fqns_ in list(shared_params_mapping.items()):313 shared_params_mapping[fqn] = cast(torch.Tensor, param_)385 shared_params_mapping=shared_params_mapping,832 if fqn in info.shared_params_mapping: