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