Searched refs:ArCrsPair (Results 1 – 2 of 2) sorted by relevance
85 struct ArCrsPair { struct91 ArCrsPair(HloInstruction* all_reduce, HloInstruction* cross_replica_sum, in ArCrsPair() argument101 absl::optional<ArCrsCombiner::ArCrsPair> MatchesArCrsPattern( argument142 absl::flat_hash_map<int64, std::vector<ArCrsPair>> all_reduce_map_;
41 absl::optional<ArCrsCombiner::ArCrsPair> ArCrsCombiner::MatchesArCrsPattern( in MatchesArCrsPattern()89 return absl::optional<ArCrsPair>(ArCrsPair(instruction, next, distance)); in MatchesArCrsPattern()