Searched refs:grouped_tiling_dims (Results 1 – 1 of 1) sorted by relevance
1388 std::vector<int64> grouped_tiling_dims = in GroupShardingOnDims() local1392 CHECK_EQ(grouped_tiling_dims[group_dims[i]] % group_dim_shards[i], 0); in GroupShardingOnDims()1394 grouped_tiling_dims[group_dims[i]] / group_dim_shards[i]; in GroupShardingOnDims()1395 grouped_tiling_dims[group_dims[i]] = group_dim_shards[i]; in GroupShardingOnDims()1417 if (Product(grouped_tiling_dims) == 1 || in GroupShardingOnDims()1419 Product(grouped_tiling_dims) == grouped_tiling_dims.back())) { in GroupShardingOnDims()1422 if (sharding.ReplicateOnLastTileDim() && grouped_tiling_dims.back() == 1) { in GroupShardingOnDims()1423 grouped_tiling_dims.pop_back(); in GroupShardingOnDims()1425 Array<int64> grouped_tiling(grouped_tiling_dims); in GroupShardingOnDims()1428 grouped_tiling_dims.size() == in GroupShardingOnDims()