commit 355ec86e256656a07b906d6408342ae1c9cdf0a7
author MLIR Team <no-reply@google.com> Sat Jun 23 18:09:09 2018 -0700
committer jpienaar <jpienaar@google.com> Fri Mar 29 12:24:58 2019 -0700
tree aaee843047df1ec4399f6ff8d3971f0eac56f0e9
parent 4c95a50715c220b993f153eee6c10cc84b95e66b

Add tensor type.

PiperOrigin-RevId: 201830793

lib/Parser/Parser.cpp

diff --git a/lib/Parser/Parser.cpp b/lib/Parser/Parser.cpp
index 11d289f..d5cc707 100644
--- a/lib/Parser/Parser.cpp
+++ b/lib/Parser/Parser.cpp
@@ -345,8 +345,9 @@
   if (!consumeIf(Token::greater))
     return (emitError("expected '>' in tensor type"), nullptr);
 
-  // FIXME: Add an IR representation for tensor types.
-  return Type::getI1(context);
+  if (isUnranked)
+    return UnrankedTensorType::get(elementType);
+  return RankedTensorType::get(dimensions, elementType);
 }
 
 /// Parse a memref type.
@@ -527,7 +528,7 @@
     return block;
   }
 private:
-  CFGFunction *function;  
+  CFGFunction *function;
   llvm::StringMap<BasicBlock*> blocksByName;
 };
 } // end anonymous namespace