Searched defs:all_reduce_map_ (Results 1 – 2 of 2) sorted by relevance
142 absl::flat_hash_map<int64, std::vector<ArCrsPair>> all_reduce_map_; variable
277 absl::flat_hash_map<int64, std::vector<HloInstruction*>> all_reduce_map_; variable