Searched refs:all_reduce_sets (Results 1 – 1 of 1) sorted by relevance
212 all_reduce_sets; in CreateComputationGroups() local216 all_reduce_sets[channel_id(hlo)].emplace_back(group_id, hlo); in CreateComputationGroups()237 if (all_reduce_sets.count(arid) == 0) { in CreateComputationGroups()244 for (const auto& hlo : all_reduce_sets[arid]) { in CreateComputationGroups()249 all_reduce_sets.erase(arid); in CreateComputationGroups()251 CHECK(all_reduce_sets.empty()); in CreateComputationGroups()