Searched defs:grouped_sharding (Results 1 – 1 of 1) sorted by relevance
1482 HloSharding UngroupSharding(const GroupedSharding& grouped_sharding) { in UngroupSharding()1529 GroupedSharding AlignGroupsWith(GroupedSharding grouped_sharding, in AlignGroupsWith()1603 Shape GetPerGroupBaseShape(const GroupedSharding& grouped_sharding, in GetPerGroupBaseShape()