| 3227 | } |
| 3228 | |
| 3229 | XlaOp Add(const XlaOp lhs, const XlaOp rhs, |
| 3230 | absl::Span<const int64> broadcast_dimensions) { |
| 3231 | return lhs.builder()->BinaryOp(HloOpcode::kAdd, lhs, rhs, |
| 3232 | broadcast_dimensions); |
| 3233 | } |
| 3234 | |
| 3235 | XlaOp Sub(const XlaOp lhs, const XlaOp rhs, |
| 3236 | absl::Span<const int64> broadcast_dimensions) { |