Searched defs:GroupedSharding (Results 1 – 1 of 1) sorted by relevance
297 struct GroupedSharding { struct298 GroupedSharding(std::vector<std::vector<int64>> device_groups, in GroupedSharding() argument314 // Creates a GroupedSharding for a tiled sharding with group dim shard sizes. argument