From a327fa069cfe9ac54986f00586309124ed037926 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <nobody@tensorflow.org>
Date: Wed, 6 Jan 2016 12:54:45 -0800
Subject: Allow slicing up to rank 6 tensors. Change: 111527673

---
 tensorflow/core/kernels/slice_op.cc | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'tensorflow/core/kernels/slice_op.cc')

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);
-- 
cgit v1.2.3