| 3295 | } |
| 3296 | |
| 3297 | XlaOp ShiftLeft(const XlaOp lhs, const XlaOp rhs, |
| 3298 | absl::Span<const int64> broadcast_dimensions) { |
| 3299 | return lhs.builder()->BinaryOp(HloOpcode::kShiftLeft, lhs, rhs, |
| 3300 | broadcast_dimensions); |
| 3301 | } |
| 3302 | |
| 3303 | XlaOp ShiftRightArithmetic(const XlaOp lhs, const XlaOp rhs, |
| 3304 | absl::Span<const int64> broadcast_dimensions) { |