commit 94379dd16d9b276a6a007f595bbbe56dffb1a281
author A. Unique TensorFlower <gardener@tensorflow.org> Wed Apr 13 10:12:56 2022 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Wed Apr 13 10:16:34 2022 -0700
tree fa72d0a77f44824a4fd55692ce6248016de6b509
parent c90e864e1b8d735068d93bb8fa46eb60c428f20b

Add a method to describe numerical classifications for comparisons.

PiperOrigin-RevId: 441513245

tensorflow/compiler/xla/comparison_util.h

diff --git a/tensorflow/compiler/xla/comparison_util.h b/tensorflow/compiler/xla/comparison_util.h
index 5688c91..2189121 100644
--- a/tensorflow/compiler/xla/comparison_util.h
+++ b/tensorflow/compiler/xla/comparison_util.h
@@ -134,6 +134,13 @@
     return primitive_type_ == PrimitiveType::U32 && IsTotalOrder();
   }
 
+  inline bool IsIntegralPrimitiveType() const {
+    return primitive_util::IsIntegralType(primitive_type_);
+  }
+  inline bool IsFloatingPointPrimitiveType() const {
+    return primitive_util::IsFloatingPointType(primitive_type_);
+  }
+
   // Returns whether (a dir a) is always true for this comparison.
   bool IsReflexive() const;