diff options
Diffstat (limited to 'tensorflow/core/kernels/cwise_op_greater_equal.cc')
| -rw-r--r-- | tensorflow/core/kernels/cwise_op_greater_equal.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/cwise_op_greater_equal.cc b/tensorflow/core/kernels/cwise_op_greater_equal.cc index ac21528256..8f0c483aec 100644 --- a/tensorflow/core/kernels/cwise_op_greater_equal.cc +++ b/tensorflow/core/kernels/cwise_op_greater_equal.cc @@ -35,7 +35,7 @@ REGISTER_KERNEL_BUILDER(Name("GreaterEqual") #endif #ifdef TENSORFLOW_USE_SYCL -REGISTER(BinaryOp, SYCL, "GreaterEqual", functor::greater_equal, float); +REGISTER2(BinaryOp, SYCL, "GreaterEqual", functor::greater_equal, float, double); REGISTER_KERNEL_BUILDER(Name("GreaterEqual") .Device(DEVICE_SYCL) |