Add convenience method to c++ api for tensor and tensor buffer to check type directly. This alleviates the minor annoyance of checking the expected before comparison and converting a c type.

LiteRT-PiperOrigin-RevId: 775400125

Author: LukeBoyer

litert/cc/litert_model.h

@@ -61,6 +61,10 @@ class RankedTensorType {
     return ElementType() == other.ElementType() && Layout() == other.Layout();
   }
 
+  bool operator!=(const RankedTensorType& other) const {
+    return !(*this == other);
+  }
+
   ElementType ElementType() const { return element_type_; }
 
   const Layout& Layout() const { return layout_; }
@@ -105,6 +109,16 @@ class Tensor : public internal::NonOwnedHandle<LiteRtTensor> {
     }
   }
 
+  bool HasType(const RankedTensorType& type) const {
+    auto t = RankedTensorType();
+    return t && *t == type;
+  }
+
+  bool HasType(const LiteRtRankedTensorType& type) const {
+    auto t = RankedTensorType();
+    return t && *t == ::litert::RankedTensorType(type);
+  }
+
   LiteRtTensorTypeId TypeId() const {
     LiteRtTensorTypeId type_id;
     internal::AssertOk(LiteRtGetTensorTypeId, Get(), &type_id);

litert/cc/litert_tensor_buffer.h

@@ -209,6 +209,15 @@ class TensorBuffer
     return RankedTensorType(tensor_type);
   }
 
+  bool HasType(const RankedTensorType& type) const {
+    auto t = TensorType();
+    return t && *t == type;
+  }
+
+  bool HasType(const LiteRtRankedTensorType& type) const {
+    auto t = TensorType();
+    return t && *t == ::litert::RankedTensorType(type);
+  }
   // Returns the size of the underlying H/W tensor buffer. This size can be
   // different to the PackedSize() if there is stride and padding exists.
   Expected<size_t> Size() const {