commit 36bba466bd4200b48f3ce7e1cf2539c5d2932251
author Geoffrey Martin-Noble <gcmn@google.com> Tue May 21 15:45:25 2019 -0700
committer Mehdi Amini <joker.eph@gmail.com> Sat Jun 01 19:55:43 2019 -0700
tree b86fc5e4fdbe6268a7a6079d9b98e8afa74d4e4d
parent 05a0d165ed427d4c9d190a4faff1ed464e30a007

    Make static shape tensor container match other container types

    Suffix with Of and take a list of types. Also give a better description that includes the element type information.

--

PiperOrigin-RevId: 249341159

include/mlir/IR/OpBase.td

diff --git a/include/mlir/IR/OpBase.td b/include/mlir/IR/OpBase.td
index e49e7dc..15477f5 100644
--- a/include/mlir/IR/OpBase.td
+++ b/include/mlir/IR/OpBase.td
@@ -370,12 +370,12 @@
 
 def AnyTensor : TensorOf<[AnyType]>;
 
-// TODO(b/130807343) Fix description to contain element information.
-class StaticShapeTensor<Type t>
-    : Type<And<[ TensorOf<[t]>.predicate, HasStaticShapePred ]>,
-           "statically shaped tensor">;
+// TODO(b/130064155) Have an easy way to add another constraint to a type.
+class StaticShapeTensorOf<list<Type> allowedTypes>
+    : Type<And<[TensorOf<allowedTypes>.predicate, HasStaticShapePred]>,
+           "statically shaped " # TensorOf<allowedTypes>.description>;
 
-def AnyStaticShapeTensor : StaticShapeTensor<AnyType>;
+def AnyStaticShapeTensor : StaticShapeTensorOf<[AnyType]>;
 
 def I1Tensor   : TensorOf<[I1]>;
 def I8Tensor   : TensorOf<[I8]>;