Searched refs:crs_reserved_map_ (Results 1 – 2 of 2) sorted by relevance
253 auto it = crs_reserved_map_.find(pair.crs); in GroupAllReducesById()254 if (it != crs_reserved_map_.end()) { in GroupAllReducesById()269 crs_reserved_map_[pair.crs] = ar_id; in GroupAllReducesById()284 crs_reserved_map_[pair.crs] = ar_id; in GroupAllReducesById()
147 absl::flat_hash_map<HloInstruction*, int64> crs_reserved_map_; variable