| 3307 | } |
| 3308 | |
| 3309 | XlaOp ShiftRightLogical(const XlaOp lhs, const XlaOp rhs, |
| 3310 | absl::Span<const int64> broadcast_dimensions) { |
| 3311 | return lhs.builder()->BinaryOp(HloOpcode::kShiftRightLogical, lhs, rhs, |
| 3312 | broadcast_dimensions); |
| 3313 | } |
| 3314 | |
| 3315 | XlaOp Reduce(const XlaOp operand, const XlaOp init_value, |
| 3316 | const XlaComputation& computation, |