From fe59ba19afbd8041027d37d19c7d23a337e7248e Mon Sep 17 00:00:00 2001
From: Geoffrey Martin-Noble <gcmn@google.com>
Date: Wed, 10 Apr 2019 17:16:19 -0700
Subject: [PATCH]     Add support for statically shaped and typed tensors

--

PiperOrigin-RevId: 242972594
---
 mlir/include/mlir/IR/OpBase.td | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/mlir/include/mlir/IR/OpBase.td b/mlir/include/mlir/IR/OpBase.td
index 52a4a1e..8fd2e00 100644
--- 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>;
-- 
2.7.4