| 2985 | } |
| 2986 | |
| 2987 | XlaOp Collapse(const XlaOp operand, absl::Span<const int64> dimensions) { |
| 2988 | return operand.builder()->Collapse(operand, dimensions); |
| 2989 | } |
| 2990 | |
| 2991 | XlaOp Slice(const XlaOp operand, absl::Span<const int64> start_indices, |
| 2992 | absl::Span<const int64> limit_indices, |