Searched refs:group_level_tile (Results 1 – 1 of 1) sorted by relevance
1715 Array<int64> group_level_tile(group_level_tile_dims); in PerGroupSliceFromReplicated() local1716 group_level_tile.Each([&](absl::Span<const int64> indices, int64* group) { in PerGroupSliceFromReplicated()1719 *group *= group_level_tile.dim(dim); in PerGroupSliceFromReplicated()1723 auto group_level_sharding = HloSharding::Tile(group_level_tile); in PerGroupSliceFromReplicated()