Searched defs:GroupShardingOnDims (Results 1 – 1 of 1) sorted by relevance
1625 GroupedSharding GroupShardingOnDims(const HloSharding& sharding, in GroupShardingOnDims() function1633 GroupedSharding GroupShardingOnDims(const HloSharding& sharding, in GroupShardingOnDims() function