Searched refs:alloc_all_gather_outputs (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/distributed/_composable/fsdp/ | ||
| D | _fsdp_collectives.py | 268 fsdp_param.alloc_all_gather_outputs() |
| D | _fsdp_param.py | 603 def alloc_all_gather_outputs(self) -> None: member in FSDPParam |