| 3552 | } |
| 3553 | |
| 3554 | XlaOp Gather(const XlaOp input, const XlaOp start_indices, |
| 3555 | const GatherDimensionNumbers& dimension_numbers, |
| 3556 | absl::Span<const int64> slice_sizes, bool indices_are_sorted) { |
| 3557 | return input.builder()->Gather(input, start_indices, dimension_numbers, |
| 3558 | slice_sizes, indices_are_sorted); |
| 3559 | } |
| 3560 | |
| 3561 | XlaOp Scatter(const XlaOp input, const XlaOp scatter_indices, |
| 3562 | const XlaOp updates, const XlaComputation& update_computation, |