| 3072 | } |
| 3073 | |
| 3074 | XlaOp Compare(const XlaOp lhs, const XlaOp rhs, |
| 3075 | absl::Span<const int64> broadcast_dimensions, |
| 3076 | ComparisonDirection direction) { |
| 3077 | return lhs.builder()->BinaryOp(HloOpcode::kCompare, lhs, rhs, |
| 3078 | broadcast_dimensions, direction); |
| 3079 | } |
| 3080 | |
| 3081 | XlaOp Dot(const XlaOp lhs, const XlaOp rhs, |
| 3082 | const PrecisionConfig* precision_config) { |