diff options
Diffstat (limited to 'tensorflow/core/kernels/cwise_op_invert.cc')
| -rw-r--r-- | tensorflow/core/kernels/cwise_op_invert.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/cwise_op_invert.cc b/tensorflow/core/kernels/cwise_op_invert.cc index df2c02e42e..c84ee6894e 100644 --- a/tensorflow/core/kernels/cwise_op_invert.cc +++ b/tensorflow/core/kernels/cwise_op_invert.cc @@ -20,7 +20,7 @@ REGISTER6(UnaryOp, CPU, "Invert", functor::invert, int8, int16, int32, int64, uint8, uint16); #ifdef TENSORFLOW_USE_SYCL -REGISTER6(UnaryOp, SYCL, "Invert", functor::invert, int8, int16, int32, int64, +REGISTER(UnaryOp, SYCL, "Invert", functor::invert, int8, int16, int32, int64, uint8, uint16); #endif // TENSORFLOW_USE_SYCL |