aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/kernels/transpose_functor_cpu.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/core/kernels/transpose_functor_cpu.cc')
-rw-r--r--tensorflow/core/kernels/transpose_functor_cpu.cc1
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/transpose_functor_cpu.cc b/tensorflow/core/kernels/transpose_functor_cpu.cc
index 30b82f1843..3681b9a129 100644
--- a/tensorflow/core/kernels/transpose_functor_cpu.cc
+++ b/tensorflow/core/kernels/transpose_functor_cpu.cc
@@ -127,6 +127,7 @@ Status DoTranspose<SYCLDevice>(const SYCLDevice& d, const Tensor& in,
switch (in.dtype()) {
case DT_FLOAT:
+ case DT_DOUBLE:
case DT_INT32:
internal::Transpose<SYCLDevice, uint32>(d, in, perm, out);
break;