Searched defs:FSDPParamInfo (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/distributed/fsdp/ |
| H A D | _optim_utils.py | 67 class FSDPParamInfo: class
134 fsdp_param_info: FSDPParamInfo,
193 fsdp_param_info: FSDPParamInfo,
263 fsdp_param_info: FSDPParamInfo,
382 fsdp_param_info: FSDPParamInfo,
562 fsdp_param_info: FSDPParamInfo,
1199 fqn_to_fsdp_param_info: Dict[str, FSDPParamInfo], argument
1348 fsdp_param_info: FSDPParamInfo,
1431 fsdp_param_info: FSDPParamInfo,
1503 fsdp_param_info: FSDPParamInfo,
[all …]
|