diff options
Diffstat (limited to 'tensorflow/core/kernels/cast_op_impl_uint16.cc')
-rw-r--r-- | tensorflow/core/kernels/cast_op_impl_uint16.cc | 9 |
1 files changed, 3 insertions, 6 deletions
diff --git a/tensorflow/core/kernels/cast_op_impl_uint16.cc b/tensorflow/core/kernels/cast_op_impl_uint16.cc index cd829bae2a..3aebbdc1f3 100644 --- a/tensorflow/core/kernels/cast_op_impl_uint16.cc +++ b/tensorflow/core/kernels/cast_op_impl_uint16.cc @@ -20,15 +20,13 @@ namespace tensorflow { typedef Eigen::ThreadPoolDevice CPUDevice; typedef Eigen::GpuDevice GPUDevice; -std::function<void(OpKernelContext*, const Tensor&, Tensor*)> -GetCpuCastFromUint16(DataType dst_dtype) { +CastFunctorType GetCpuCastFromUint16(DataType dst_dtype) { CURRY_TYPES3(CAST_CASE, CPUDevice, uint16); return nullptr; } #if GOOGLE_CUDA -std::function<void(OpKernelContext*, const Tensor&, Tensor*)> -GetGpuCastFromUint16(DataType dst_dtype) { +CastFunctorType GetGpuCastFromUint16(DataType dst_dtype) { CURRY_TYPES3_NO_BF16(CAST_CASE, GPUDevice, uint16); return nullptr; } @@ -36,8 +34,7 @@ GetGpuCastFromUint16(DataType dst_dtype) { #ifdef TENSORFLOW_USE_SYCL typedef Eigen::SyclDevice SYCLDevice; -std::function<void(OpKernelContext*, const Tensor&, Tensor*)> -GetSyclCastFromUint16(DataType dst_dtype) { +CastFunctorType GetSyclCastFromUint16(DataType dst_dtype) { CURRY_TYPES3_NO_HALF(CAST_CASE, SYCLDevice, uint16); return nullptr; } |