Searched defs:group_dim_shards (Results 1 – 1 of 1) sorted by relevance
1380 std::vector<int64> group_dim_shards(group_dims.size(), 1); in GroupShardingOnDims() local1386 absl::Span<const int64> group_dim_shards) { in GroupShardingOnDims()