Searched refs:MatchReductionComputation (Results 1 – 4 of 4) sorted by relevance
39 static absl::optional<ReductionKind> MatchReductionComputation( in MatchReductionComputation() function91 auto reduction_kind = MatchReductionComputation(op); in GetNcclAllReduceConfig()107 return operands_are_supported && MatchReductionComputation(op).has_value(); in CanImplement()
22 absl::optional<ReductionKind> MatchReductionComputation( in MatchReductionComputation() function
37 absl::optional<ReductionKind> MatchReductionComputation(
1130 if (!MatchReductionComputation(crs->to_apply()).has_value()) { in HandleAllReduceMultipleReplica()1190 static_cast<int32>(*MatchReductionComputation(crs->to_apply()))), in HandleAllReduceMultipleReplica()