Searched refs:group_level_sharding (Results 1 – 1 of 1) sorted by relevance
1723 auto group_level_sharding = HloSharding::Tile(group_level_tile); in PerGroupSliceFromReplicated() local1725 replicated, group_level_sharding, b); in PerGroupSliceFromReplicated()1727 MakePartitionedShape(replicated->shape(), group_level_sharding); in PerGroupSliceFromReplicated()1730 MakePartitionOffsets(replicated->shape(), group_level_sharding, group_id, in PerGroupSliceFromReplicated()