Home
last modified time | relevance | path

Searched refs:prev_ar_id (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dar_crs_combiner.cc255 auto prev_ar_id = it->second; in GroupAllReducesById() local
260 CHECK_NE(prev_ar_id, ar_id); in GroupAllReducesById()
261 auto prev_pair = all_reduce_map_[prev_ar_id].back(); in GroupAllReducesById()
266 all_reduce_map_.erase(prev_ar_id); in GroupAllReducesById()
267 discarded_ar_ids.insert(prev_ar_id); in GroupAllReducesById()