diff options
Diffstat (limited to 'tensorflow/core/kernels/cwise_op_sigmoid.cc')
-rw-r--r-- | tensorflow/core/kernels/cwise_op_sigmoid.cc | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/cwise_op_sigmoid.cc b/tensorflow/core/kernels/cwise_op_sigmoid.cc index cc1f9b8f03..a76a088ac8 100644 --- a/tensorflow/core/kernels/cwise_op_sigmoid.cc +++ b/tensorflow/core/kernels/cwise_op_sigmoid.cc @@ -23,6 +23,9 @@ REGISTER5(UnaryOp, CPU, "Sigmoid", functor::sigmoid, float, Eigen::half, double, REGISTER3(UnaryOp, GPU, "Sigmoid", functor::sigmoid, float, Eigen::half, double); #endif +#ifdef TENSORFLOW_USE_SYCL +REGISTER(UnaryOp, SYCL, "Sigmoid", functor::sigmoid, float); +#endif // TENSORFLOW_USE_SYCL REGISTER5(SimpleBinaryOp, CPU, "SigmoidGrad", functor::sigmoid_grad, float, Eigen::half, double, complex64, complex128); @@ -30,5 +33,8 @@ REGISTER5(SimpleBinaryOp, CPU, "SigmoidGrad", functor::sigmoid_grad, float, REGISTER3(SimpleBinaryOp, GPU, "SigmoidGrad", functor::sigmoid_grad, float, Eigen::half, double); #endif +#ifdef TENSORFLOW_USE_SYCL +REGISTER(SimpleBinaryOp, SYCL, "SigmoidGrad", functor::sigmoid_grad, float); +#endif // TENSORFLOW_USE_SYCL } // namespace tensorflow |