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