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