Searched refs:ar_id (Results 1 – 1 of 1) sorted by relevance
249 int64 ar_id = *(instruction->all_reduce_id()); in GroupAllReducesById() local250 if (discarded_ar_ids.find(ar_id) != discarded_ar_ids.end()) { in GroupAllReducesById()259 CHECK(all_reduce_map_.find(ar_id) == all_reduce_map_.end()); in GroupAllReducesById()260 CHECK_NE(prev_ar_id, ar_id); in GroupAllReducesById()268 all_reduce_map_[ar_id].push_back(pair); in GroupAllReducesById()269 crs_reserved_map_[pair.crs] = ar_id; in GroupAllReducesById()273 discarded_ar_ids.insert(ar_id); in GroupAllReducesById()276 if (all_reduce_map_.find(ar_id) != all_reduce_map_.end()) { in GroupAllReducesById()277 int64 prev_distance = all_reduce_map_[ar_id].back().distance; in GroupAllReducesById()283 all_reduce_map_[ar_id].push_back(pair); in GroupAllReducesById()[all …]