diff options
Diffstat (limited to 'tensorflow/core/kernels/cwise_op_neg.cc')
| -rw-r--r-- | tensorflow/core/kernels/cwise_op_neg.cc | 17 |
1 files changed, 2 insertions, 15 deletions
diff --git a/tensorflow/core/kernels/cwise_op_neg.cc b/tensorflow/core/kernels/cwise_op_neg.cc index eb7e3764d9..536891b548 100644 --- a/tensorflow/core/kernels/cwise_op_neg.cc +++ b/tensorflow/core/kernels/cwise_op_neg.cc @@ -19,27 +19,14 @@ namespace tensorflow { REGISTER7(UnaryOp, CPU, "Neg", functor::neg, float, Eigen::half, double, int32, complex64, int64, complex128); -#if TENSORFLOW_USE_SYCL -#define REGISTER_SYCL_KERNEL(TYPE) \ - REGISTER_KERNEL_BUILDER( \ - Name("Neg") \ - .Device(DEVICE_SYCL) \ - .TypeConstraint<TYPE>("T"), \ - UnaryOp<SYCLDevice, functor::neg<TYPE>>); -REGISTER_SYCL_KERNEL(float); -REGISTER_SYCL_KERNEL(double); - -// A special GPU kernel for int32. -// TODO(b/25387198): Also enable int32 in device memory. This kernel -// registration requires all int32 inputs and outputs to be in host memory. +#ifdef TENSORFLOW_USE_SYCL +REGISTER3(UnaryOp, SYCL, "Neg", functor::neg, float, double, int64); REGISTER_KERNEL_BUILDER(Name("Neg") .Device(DEVICE_SYCL) .HostMemory("x") .HostMemory("y") .TypeConstraint<int32>("T"), UnaryOp<CPUDevice, functor::neg<int32>>); - -#undef REGISTER_SYCL_KERNEL #endif // TENSORFLOW_USE_SYCL #if GOOGLE_CUDA |