diff options
Diffstat (limited to 'tensorflow/core/kernels/cwise_op_expm1.cc')
| -rw-r--r-- | tensorflow/core/kernels/cwise_op_expm1.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/cwise_op_expm1.cc b/tensorflow/core/kernels/cwise_op_expm1.cc index 5573c2bcc2..4f72308006 100644 --- a/tensorflow/core/kernels/cwise_op_expm1.cc +++ b/tensorflow/core/kernels/cwise_op_expm1.cc @@ -22,6 +22,6 @@ REGISTER5(UnaryOp, CPU, "Expm1", functor::expm1, float, Eigen::half, double, REGISTER3(UnaryOp, GPU, "Expm1", functor::expm1, float, Eigen::half, double); #endif #ifdef TENSORFLOW_USE_SYCL -REGISTER(UnaryOp, SYCL, "Expm1", functor::expm1, float); +REGISTER2(UnaryOp, SYCL, "Expm1", functor::expm1, float, double); #endif // TENSORFLOW_USE_SYCL } // namespace tensorflow |