diff --git a/mlir/include/mlir/Interfaces/InferTypeOpInterface.h b/mlir/include/mlir/Interfaces/InferTypeOpInterface.h --- a/mlir/include/mlir/Interfaces/InferTypeOpInterface.h +++ b/mlir/include/mlir/Interfaces/InferTypeOpInterface.h @@ -76,6 +76,10 @@ Attribute attr; }; +/// Range of values and shapes (corresponding effectively to Shapes dialect's +/// ValueShape type concept). +using ValueShapeRange = ValueRange; + namespace detail { // Helper function to infer return tensor returns types given element and shape // inference function. diff --git a/mlir/include/mlir/Interfaces/InferTypeOpInterface.td b/mlir/include/mlir/Interfaces/InferTypeOpInterface.td --- a/mlir/include/mlir/Interfaces/InferTypeOpInterface.td +++ b/mlir/include/mlir/Interfaces/InferTypeOpInterface.td @@ -93,7 +93,7 @@ /*methodName=*/"inferReturnTypeComponents", /*args=*/(ins "::mlir::MLIRContext*":$context, "::mlir::Optional<::mlir::Location>":$location, - "::mlir::ValueRange":$operands, + "::mlir::ValueShapeRange":$operands, "::mlir::DictionaryAttr":$attributes, "::mlir::RegionRange":$regions, "::mlir::SmallVectorImpl<::mlir::ShapedTypeComponents>&":