Searched defs:_unshard (Results 1 – 3 of 3) sorted by relevance
273 def _unshard( function
2054 def _unshard(self, async_op: bool = False): member in FullyShardedDataParallel
923 def _unshard(stage_index: int): function