diff options
Diffstat (limited to 'tensorflow/core/kernels/cwise_op_round.cc')
| -rw-r--r-- | tensorflow/core/kernels/cwise_op_round.cc | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/cwise_op_round.cc b/tensorflow/core/kernels/cwise_op_round.cc index 0457f3931d..7a4482dbb2 100644 --- a/tensorflow/core/kernels/cwise_op_round.cc +++ b/tensorflow/core/kernels/cwise_op_round.cc @@ -18,6 +18,14 @@ limitations under the License. namespace tensorflow { REGISTER5(UnaryOp, CPU, "Round", functor::round, Eigen::half, float, double, int32, int64); + +#ifdef TENSORFLOW_USE_SYCL +REGISTER(UnaryOp, SYCL, "Round", functor::round, float); +namespace functor { +DEFINE_UNARY1(round, float); +} // namespace functor +#endif + #if GOOGLE_CUDA REGISTER5(UnaryOp, GPU, "Round", functor::round, Eigen::half, float, double, int32, int64); |