Add support for statically shaped and typed tensors

--

PiperOrigin-RevId: 242972594

diff --git a/mlir/include/mlir/IR/OpBase.td b/mlir/include/mlir/IR/OpBase.td
index 52a4a1e..8fd2e00 100644 (file)
--- a/mlir/include/mlir/IR/OpBase.td
+++ b/mlir/include/mlir/IR/OpBase.td
@@ -315,6 +315,10 @@ class TypedTensor<Type t>
                     "{0}.cast<TensorType>().getElementType()",
                     "tensor">;
 
+class TypedStaticShapeTensor<Type t>
+    : Type<AllOf<[ TypedTensor<t>.predicate, IsStaticShapeTensorTypePred ]>,
+           "statically shaped tensor">;
+
 def I1Tensor  : TypedTensor<I1>;
 def I8Tensor  : TypedTensor<I8>;
 def I16Tensor : TypedTensor<I16>;