Add uint16, uint32, uint64 support for tf.math.not_equal

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
2020-04-06 22:52:43 +00:00 · 2020-04-06 22:52:43 +00:00 · 67ef513917
parent 928ff3e27b
commit 67ef513917
1 changed files with 1 additions and 0 deletions
--- 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);