Home
last modified time | relevance | path

Searched defs:all_reduce_map_ (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dar_crs_combiner.h183 absl::flat_hash_map<int64_t, std::vector<ArCrsPair>> all_reduce_map_; variable
H A Dhlo_module_group_metadata.h276 absl::flat_hash_map<int64_t, std::vector<HloInstruction*>> all_reduce_map_; variable