| 3461 | return operand.builder()->UnaryOp(HloOpcode::kSqrt, operand); |
| 3462 | } |
| 3463 | XlaOp Rsqrt(const XlaOp operand) { |
| 3464 | return operand.builder()->UnaryOp(HloOpcode::kRsqrt, operand); |
| 3465 | } |
| 3466 | |
| 3467 | XlaOp Pow(const XlaOp lhs, const XlaOp rhs, |
| 3468 | absl::Span<const int64> broadcast_dimensions) { |