1 files changed, 12 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/transpose_functor_cpu.cc b/tensorflow/core/kernels/transpose_functor_cpu.cc
index 41b73fdaf4..6594f7ee7b 100644
--- a/tensorflow/core/kernels/transpose_functor_cpu.cc
+++ b/tensorflow/core/kernels/transpose_functor_cpu.cc
@@ -88,6 +88,18 @@ struct Transpose<CPUDevice, T, conjugate> {
         internal::TransposeUsingEigen<CPUDevice, T, 5>(d, in, perm, conjugate,
                                                        out);
         break;
+      case 6:
+	internal::TransposeUsingEigen<CPUDevice, T, 6>(d, in, perm, conjugate,
+						       out);
+	break;
+      case 7:
+	internal::TransposeUsingEigen<CPUDevice, T, 7>(d, in, perm, conjugate,
+						       out);
+	break;
+      case 8:
+        internal::TransposeUsingEigen<CPUDevice, T, 8>(d, in, perm, conjugate,
+						       out);
+	break;
       default:
         TransposeSimple<T, conjugate>(d, in, perm, out);
         break;