maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

Add support for statically shaped and typed tensors 

--

PiperOrigin-RevId: 242972594
This commit is contained in:
Geoffrey Martin-Noble 2019-04-10 17:16:19 -07:00 committed by Mehdi Amini
parent d468eaccfc
commit fe59ba19af
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mlir/include/mlir/IR/OpBase.td
View File

@ -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>;