(x, start_dims, limit_dims, strides)
| 391 | |
| 392 | |
| 393 | def slice(x, start_dims, limit_dims, strides): |
| 394 | spec = [ |
| 395 | _slice(start, limit, stride) |
| 396 | for (start, limit, stride) in zip(start_dims, limit_dims, strides) |
| 397 | ] |
| 398 | return x[tuple(spec)] |
| 399 | |
| 400 | |
| 401 | sort = gen_xla_ops.xla_sort |