diff options
| author |  luke <luke@codeplay.com> | 2016-10-28 11:27:29 +0100 |
|---|---|---|
| committer | luke <luke@codeplay.com> | 2016-10-28 11:27:29 +0100 |
| commit | 0b4ce7841aa2d18c8546cfe75ef2c58b89864996 (patch) | |
| tree | 26ae2cbf89bdb3f30a204816c61c2aeb62ebd32e /tensorflow/core/kernels/cwise_op_round.cc | |
| parent | 9401110a9d8545c40fce638e4c63fcbd2668d0fc (diff) | |
Added #ifdef guard for SYCL related code.
Diffstat (limited to 'tensorflow/core/kernels/cwise_op_round.cc')
| -rw-r--r-- | tensorflow/core/kernels/cwise_op_round.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/cwise_op_round.cc b/tensorflow/core/kernels/cwise_op_round.cc index c87157f2d3..7a4482dbb2 100644 --- a/tensorflow/core/kernels/cwise_op_round.cc +++ b/tensorflow/core/kernels/cwise_op_round.cc @@ -19,10 +19,12 @@ 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, |