| 3251 | } |
| 3252 | |
| 3253 | XlaOp Rem(const XlaOp lhs, const XlaOp rhs, |
| 3254 | absl::Span<const int64> broadcast_dimensions) { |
| 3255 | return lhs.builder()->BinaryOp(HloOpcode::kRemainder, lhs, rhs, |
| 3256 | broadcast_dimensions); |
| 3257 | } |
| 3258 | |
| 3259 | XlaOp Max(const XlaOp lhs, const XlaOp rhs, |
| 3260 | absl::Span<const int64> broadcast_dimensions) { |