diff options
Diffstat (limited to 'tensorflow/core/kernels/cwise_op_add_2.cc')
-rw-r--r-- | tensorflow/core/kernels/cwise_op_add_2.cc | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/tensorflow/core/kernels/cwise_op_add_2.cc b/tensorflow/core/kernels/cwise_op_add_2.cc index 5dea00e95c..5d3385b0ed 100644 --- a/tensorflow/core/kernels/cwise_op_add_2.cc +++ b/tensorflow/core/kernels/cwise_op_add_2.cc @@ -22,11 +22,10 @@ namespace tensorflow { // sharded files, only make its register calls when not __ANDROID_TYPES_SLIM__. #if !defined(__ANDROID_TYPES_SLIM__) -REGISTER6(BinaryOp, CPU, "Add", functor::add, int8, int16, complex64, - uint8, complex128, string); +REGISTER5(BinaryOp, CPU, "Add", functor::add, int8, int16, complex64, + complex128, string); #if GOOGLE_CUDA -REGISTER4(BinaryOp, GPU, "Add", functor::add, uint8, int64, complex64, - complex128); +REGISTER3(BinaryOp, GPU, "Add", functor::add, int64, complex64, complex128); #endif // GOOGLE_CUDA #endif // !defined(__ANDROID_TYPES_SLIM__) |