Searched refs:new_num_shards_per_dim (Results 1 – 1 of 1) sorted by relevance
83 const std::vector<int>& new_num_shards_per_dim) { in ExpandShardVector() argument87 auto ExpandShard = [shard_vec, new_num_shards_per_dim]( in ExpandShardVector()91 int new_dim_size = new_num_shards_per_dim[dim_ind]; in ExpandShardVector()109 for (int dim_ind = 0; dim_ind < new_num_shards_per_dim.size(); ++dim_ind) { in ExpandShardVector()124 expanded_shard_vec.num_shards_per_dim = new_num_shards_per_dim; in ExpandShardVector()