aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/kernels/slice_op_cpu_impl.h
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/core/kernels/slice_op_cpu_impl.h')
-rw-r--r--tensorflow/core/kernels/slice_op_cpu_impl.h12
1 files changed, 12 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/slice_op_cpu_impl.h b/tensorflow/core/kernels/slice_op_cpu_impl.h
index 0b0700ec36..a70805658e 100644
--- a/tensorflow/core/kernels/slice_op_cpu_impl.h
+++ b/tensorflow/core/kernels/slice_op_cpu_impl.h
@@ -34,6 +34,18 @@ DEFINE_CPU_KERNELS(bfloat16);
#undef DEFINE_CPU_KERNELS
+#ifdef TENSORFLOW_USE_SYCL
+using SyclDevice = Eigen::SyclDevice;
+
+#define DEFINE_SYCL_KERNELS(T) \
+ template struct functor::Slice<SyclDevice, T, CPU_PROVIDED_IXDIM>;
+
+TF_CALL_GPU_NUMBER_TYPES(DEFINE_SYCL_KERNELS);
+DEFINE_SYCL_KERNELS(int32);
+
+#undef DEFINE_SYCL_KERNELS
+#endif // TENSORFLOW_USE_SYCL
+
} // namespace tensorflow
#endif // THIRD_PARTY_TENSORFLOW_CORE_KERNELS_SLICE_OP_CPU_IMPL_H_