Searched refs:_unshard_shape (Results 1 – 1 of 1) sorted by relevance
245 def _unshard_shape(self, shape): member in ShardingPolicy299 unsharded_shapes = [self._unshard_shape(s) for s in shapes]