Searched refs:global_ranks (Results 1 – 2 of 2) sorted by relevance
379 std::vector<size_t>& global_ranks, in GetGlobalRanks() argument401 global_ranks[s] = rank; in GetGlobalRanks()406 global_ranks = context_.net.AllReduce( in GetGlobalRanks()407 global_ranks, common::ComponentSum<std::vector<size_t> >()); in GetGlobalRanks()430 const std::vector<size_t>& global_ranks, in SearchStep() argument446 if (global_ranks[s] < target_ranks[s]) { in SearchStep()450 else if (global_ranks[s] >= target_ranks[s]) { in SearchStep()525 std::vector<size_t> global_ranks(p - 1); in MainOp() local573 GetGlobalRanks(pivots, global_ranks, local_ranks, left, width); in MainOp()575 LOG << "global_ranks: " << global_ranks; in MainOp()[all …]
33 for global_ranks in name_to_global_ranks.values():34 global_ranks.sort()