Searched refs:shard_dims (Results 1 – 1 of 1) sorted by relevance
76 std::map<int64_t, int64_t> shard_dims = AllRankLocalShard(first_dim, rank_id, server_num); in LocalShard() local77 if (shard_dims.count(rank_id) == 0) { in LocalShard()80 return shard_dims[rank_id]; in LocalShard()90 std::map<int64_t, int64_t> shard_dims; in AllRankLocalShard() local92 shard_dims[i] = 0; in AllRankLocalShard()94 if (server_num != static_cast<int64_t>(shard_dims.size())) { in AllRankLocalShard()95 …N) << "Inconsistent server num " << server_num << " shard dims counter size " << shard_dims.size(); in AllRankLocalShard()100 shard_dims[server_index] = shard_dims[server_index] + 1; in AllRankLocalShard()102 if (shard_dims.count(rank_id) == 0) { in AllRankLocalShard()105 return shard_dims; in AllRankLocalShard()