diff options
Diffstat (limited to 'tensorflow/core/kernels/concat_op.cc')
| -rw-r--r-- | tensorflow/core/kernels/concat_op.cc | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/concat_op.cc b/tensorflow/core/kernels/concat_op.cc index 916bbc4996..e7848a7e26 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<SYCLDevice, type>) -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<int32>("T") @@ -248,6 +249,7 @@ REGISTER_KERNEL_BUILDER(Name("ConcatV2") .HostMemory("axis") .HostMemory("output"), ConcatV2Op<CPUDevice, int32>); + #undef REGISTER_SYCL #endif // TENSORFLOW_USE_SYCL |