diff --git a/tensorflow/core/kernels/relu_op.cc b/tensorflow/core/kernels/relu_op.cc index d70398bea57..d8d30e87e22 100644 --- a/tensorflow/core/kernels/relu_op.cc +++ b/tensorflow/core/kernels/relu_op.cc @@ -156,7 +156,7 @@ TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU_KERNELS); Name("EluGrad").Device(DEVICE_SYCL).TypeConstraint("T"), \ EluGradOp) -REGISTER_SYCL_KERNELS(float); +TF_CALL_GPU_NUMBER_TYPES_NO_HALF(REGISTER_SYCL_KERNELS); #undef REGISTER_SYCL_KERNELS #endif // TENSORFLOW_USE_SYCL diff --git a/tensorflow/core/kernels/relu_op.h b/tensorflow/core/kernels/relu_op.h index e2e0bd48dd1..365c6201a54 100644 --- a/tensorflow/core/kernels/relu_op.h +++ b/tensorflow/core/kernels/relu_op.h @@ -175,10 +175,6 @@ void EluGradOp::OperateNoTemplate(OpKernelContext* context, } // namespace tensorflow -#ifdef TENSORFLOW_USE_SYCL -#undef EIGEN_USE_SYCL -#endif // TENSORFLOW_USE_SYCL - #undef EIGEN_USE_THREADS #endif // TENSORFLOW_KERNELS_RELU_OP_H_