diff options
Diffstat (limited to 'tensorflow/core/kernels/cast_op_impl_int64.cc')
-rw-r--r-- | tensorflow/core/kernels/cast_op_impl_int64.cc | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/cast_op_impl_int64.cc b/tensorflow/core/kernels/cast_op_impl_int64.cc index b5571b19a5..c0a543708d 100644 --- a/tensorflow/core/kernels/cast_op_impl_int64.cc +++ b/tensorflow/core/kernels/cast_op_impl_int64.cc @@ -19,6 +19,9 @@ namespace tensorflow { typedef Eigen::ThreadPoolDevice CPUDevice; typedef Eigen::GpuDevice GPUDevice; +#ifdef TENSORFLOW_USE_SYCL +typedef Eigen::SyclDevice SYCLDevice; +#endif // TENSORFLOW_USE_SYCL std::function<void(OpKernelContext*, const Tensor&, Tensor*)> GetCpuCastFromInt64(DataType dst_dtype) { @@ -34,4 +37,13 @@ GetGpuCastFromInt64(DataType dst_dtype) { } #endif // GOOGLE_CUDA +#ifdef TENSORFLOW_USE_SYCL +typedef Eigen::SyclDevice SYCLDevice; +std::function<void(OpKernelContext*, const Tensor&, Tensor*)> +GetSyclCastFromInt64(DataType dst_dtype) { + CURRY_TYPES3(CAST_CASE, SYCLDevice, int64); + return nullptr; +} +#endif // TENSORFLOW_USE_SYCL + } // namespace tensorflow |