| 3233 | } |
| 3234 | |
| 3235 | XlaOp Sub(const XlaOp lhs, const XlaOp rhs, |
| 3236 | absl::Span<const int64> broadcast_dimensions) { |
| 3237 | return lhs.builder()->BinaryOp(HloOpcode::kSubtract, lhs, rhs, |
| 3238 | broadcast_dimensions); |
| 3239 | } |
| 3240 | |
| 3241 | XlaOp Mul(const XlaOp lhs, const XlaOp rhs, |
| 3242 | absl::Span<const int64> broadcast_dimensions) { |