diff options
author | A. Unique TensorFlower <nobody@tensorflow.org> | 2016-01-06 12:54:45 -0800 |
---|---|---|
committer | Vijay Vasudevan <vrv@google.com> | 2016-01-06 12:54:45 -0800 |
commit | a327fa069cfe9ac54986f00586309124ed037926 (patch) | |
tree | ac3431c02b57eab0b7cec8575b8e819aa251e36c /tensorflow/core/kernels/slice_op.cc | |
parent | 911cec6b808e841fee7c742c85bac8c4b4a3113e (diff) |
Allow slicing up to rank 6 tensors.
Change: 111527673
Diffstat (limited to 'tensorflow/core/kernels/slice_op.cc')
-rw-r--r-- | tensorflow/core/kernels/slice_op.cc | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/slice_op.cc b/tensorflow/core/kernels/slice_op.cc index 5990e64255..015b1f3465 100644 --- a/tensorflow/core/kernels/slice_op.cc +++ b/tensorflow/core/kernels/slice_op.cc @@ -175,6 +175,7 @@ class SliceOp : public OpKernel { HANDLE_DIM(3); HANDLE_DIM(4); HANDLE_DIM(5); + HANDLE_DIM(6); #undef HANDLE_DIM @@ -230,7 +231,8 @@ namespace functor { DECLARE_GPU_SPEC(T, 2); \ DECLARE_GPU_SPEC(T, 3); \ DECLARE_GPU_SPEC(T, 4); \ - DECLARE_GPU_SPEC(T, 5); + DECLARE_GPU_SPEC(T, 5); \ + DECLARE_GPU_SPEC(T, 6); TF_CALL_GPU_NUMBER_TYPES(DECLARE_FOR_N); DECLARE_FOR_N(int32); |