[mlir] Add alias for input to shaped type op interface

[jpienaar]

Summary

Range type that allows for wrapping different value & shape ranges with correspondence to Shape's ValueShape type - initially aliased to ValueRange (which corresponds to the trivial mapping from a ShapedType's Value's shape to shape). Just plain alias, before expanding.

[herhut]

mlir/include/mlir/Interfaces/InferTypeOpInterface.h

79

What does Shapes's refer to here?

[jpienaar]

Update comment.

[jpienaar]

mlir/include/mlir/Interfaces/InferTypeOpInterface.h

79

Shape dialect, I'll expand it.

mlir/include/mlir/Interfaces/InferTypeOpInterface.h

 
 private:
   ShapeStorageT dims;
   bool ranked;
   Type elementType;
   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.
 //
 // TODO: Consider generating typedefs for trait member functions if this usage
 // becomes more common.
 LogicalResult inferReturnTensorTypes(
     function_ref<LogicalResult(

mlir/include/mlir/Interfaces/InferTypeOpInterface.td

StaticInterfaceMethod<
       Unknown (e.g., unranked) shape and nullptrs for element type and attribute
       may be returned by this function while returning success. E.g., partial
       population of components is not error condition.
       }],
       /*retTy=*/"::mlir::LogicalResult",
       /*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>&":
                       $inferredReturnShapes),
       /*methodBody=*/[{}],
       /*defaultImplementation=*/[{ return ::mlir::failure(); }]
     >,
     InterfaceMethod<