From 67ef5139174c5ccfceae9219a254cdd94a2a9672 Mon Sep 17 00:00:00 2001 From: Yong Tang Date: Mon, 6 Apr 2020 22:52:43 +0000 Subject: [PATCH] Add uint16, uint32, uint64 support for tf.math.not_equal Signed-off-by: Yong Tang --- tensorflow/core/kernels/cwise_op_not_equal_to_1.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/tensorflow/core/kernels/cwise_op_not_equal_to_1.cc b/tensorflow/core/kernels/cwise_op_not_equal_to_1.cc index f207158b843..0b518d911b0 100644 --- a/tensorflow/core/kernels/cwise_op_not_equal_to_1.cc +++ b/tensorflow/core/kernels/cwise_op_not_equal_to_1.cc @@ -18,6 +18,7 @@ limitations under the License. namespace tensorflow { REGISTER7(BinaryOp, CPU, "NotEqual", functor::not_equal_to, float, Eigen::half, double, uint8, int8, int16, bfloat16); +REGISTER3(BinaryOp, CPU, "NotEqual", functor::not_equal_to, uint16, uint32, uint64); #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM REGISTER4(BinaryOp, GPU, "NotEqual", functor::not_equal_to, float, Eigen::half, double, uint8);