Searched defs:unshard (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/distributed/_composable/fsdp/ | ||
| H A D | fully_shard.py | 188 def unshard(self, async_op: bool = False) -> Optional["UnshardHandle"]: member in FSDPModule |
| H A D | _fsdp_param_group.py | 238 def unshard(self, async_op: bool = False): member in FSDPParamGroup |
| /aosp_15_r20/external/pytorch/torch/distributed/fsdp/ | ||
| H A D | _flat_param.py | 1289 def unshard(self): member in FlatParamHandle |