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