| 3559 | } |
| 3560 | |
| 3561 | XlaOp Scatter(const XlaOp input, const XlaOp scatter_indices, |
| 3562 | const XlaOp updates, const XlaComputation& update_computation, |
| 3563 | const ScatterDimensionNumbers& dimension_numbers, |
| 3564 | bool indices_are_sorted, bool unique_indices) { |
| 3565 | return input.builder()->Scatter(input, scatter_indices, updates, |
| 3566 | update_computation, dimension_numbers, |
| 3567 | indices_are_sorted, unique_indices); |
| 3568 | } |
| 3569 | |
| 3570 | void Send(const XlaOp operand, const ChannelHandle& handle) { |
| 3571 | return operand.builder()->Send(operand, handle); |