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