| 3239 | } |
| 3240 | |
| 3241 | XlaOp Mul(const XlaOp lhs, const XlaOp rhs, |
| 3242 | absl::Span<const int64> broadcast_dimensions) { |
| 3243 | return lhs.builder()->BinaryOp(HloOpcode::kMultiply, lhs, rhs, |
| 3244 | broadcast_dimensions); |
| 3245 | } |
| 3246 | |
| 3247 | XlaOp Div(const XlaOp lhs, const XlaOp rhs, |
| 3248 | absl::Span<const int64> broadcast_dimensions) { |