Add constructor to ShapedTypeComponents for unranked with element type.

Subscribers: mehdi_amini, rriddle, jpienaar, shauheen, antiagainst, nicolasvasilache, arpith-jacob, mgester, lucyrfox, liufengdb, Joonsoo, grosul1, frgossen, Kayjukh, llvm-commits

Tags: #llvm

Differential Revision: https://reviews.llvm.org/D78774

diff --git a/mlir/include/mlir/Interfaces/InferTypeOpInterface.h b/mlir/include/mlir/Interfaces/InferTypeOpInterface.h
index 0da1125..c13f2e3 100644 (file)
--- a/mlir/include/mlir/Interfaces/InferTypeOpInterface.h
+++ b/mlir/include/mlir/Interfaces/InferTypeOpInterface.h
@@ -38,7 +38,8 @@ class ShapedTypeComponents {
 public:
   /// Default construction is an unranked shape.
   ShapedTypeComponents() : ranked(false), elementType(nullptr), attr(nullptr){};
-
+  ShapedTypeComponents(Type elementType)
+      : ranked(false), elementType(elementType), attr(nullptr) {}
   template <typename Arg, typename = typename std::enable_if_t<
                               std::is_constructible<ShapeStorageT, Arg>::value>>
   ShapedTypeComponents(Arg &&arg, Type elementType = nullptr,