diff options
Diffstat (limited to 'tensorflow/core/kernels/cast_op_impl_int32.cc')
| -rw-r--r-- | tensorflow/core/kernels/cast_op_impl_int32.cc | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/tensorflow/core/kernels/cast_op_impl_int32.cc b/tensorflow/core/kernels/cast_op_impl_int32.cc index fca9cd60ec..69ed760455 100644 --- a/tensorflow/core/kernels/cast_op_impl_int32.cc +++ b/tensorflow/core/kernels/cast_op_impl_int32.cc @@ -38,10 +38,9 @@ GetGpuCastFromInt32(DataType dst_dtype) { typedef Eigen::SyclDevice SYCLDevice; std::function<void(OpKernelContext*, const Tensor&, Tensor*)> GetSyclCastFromInt32(DataType dst_dtype) { - CURRY_TYPES3(CAST_CASE, CPUDevice, int32); + CURRY_TYPES3(CAST_CASE, SYCLDevice, int32); return nullptr; } #endif // TENSORFLOW_USE_SYCL } // namespace tensorflow - |