diff --git a/tensorflow/core/kernels/concat_lib_cpu.cc b/tensorflow/core/kernels/concat_lib_cpu.cc index 9ad1e60c6cc..258ce154560 100644 --- a/tensorflow/core/kernels/concat_lib_cpu.cc +++ b/tensorflow/core/kernels/concat_lib_cpu.cc @@ -95,7 +95,7 @@ void ConcatSYCL(const Eigen::SyclDevice& d, const std::vector::ConstMatrix>>&, \ typename TTypes::Matrix* output); -TF_CALL_GPU_NUMBER_TYPES(REGISTER_SYCL) +TF_CALL_GPU_NUMBER_TYPES_NO_HALF(REGISTER_SYCL) #undef REGISTER_SYCL #endif // TENSORFLOW_USE_SYCL diff --git a/tensorflow/core/kernels/concat_op.cc b/tensorflow/core/kernels/concat_op.cc index 916bbc49963..e7848a7e260 100644 --- a/tensorflow/core/kernels/concat_op.cc +++ b/tensorflow/core/kernels/concat_op.cc @@ -232,7 +232,8 @@ REGISTER_KERNEL_BUILDER(Name("ConcatV2") .HostMemory("axis"), \ ConcatV2Op) -TF_CALL_GPU_NUMBER_TYPES(REGISTER_SYCL); +TF_CALL_GPU_NUMBER_TYPES_NO_HALF(REGISTER_SYCL); + REGISTER_KERNEL_BUILDER(Name("Concat") .Device(DEVICE_SYCL) .TypeConstraint("T") @@ -248,6 +249,7 @@ REGISTER_KERNEL_BUILDER(Name("ConcatV2") .HostMemory("axis") .HostMemory("output"), ConcatV2Op); + #undef REGISTER_SYCL #endif // TENSORFLOW_USE_SYCL