14 #ifndef MLIR_DIALECT_SHAPE_IR_SHAPE_H
15 #define MLIR_DIALECT_SHAPE_IR_SHAPE_H
30 #define GET_TYPEDEF_CLASSES
31 #include "mlir/Dialect/Shape/IR/ShapeOpsTypes.h.inc"
34 class PatternRewriter;
40 int64_t rank = ShapedType::kDynamic);
46 LogicalResult
getShapeVec(Value input, SmallVectorImpl<int64_t> &shapeValues);
51 #define GET_OP_CLASSES
52 #include "mlir/Dialect/Shape/IR/ShapeOps.h.inc"
54 #include "mlir/Dialect/Shape/IR/ShapeOpsDialect.h.inc"
bool isExtentTensorType(Type)
LogicalResult getShapeVec(Value input, SmallVectorImpl< int64_t > &shapeValues)
RankedTensorType getExtentTensorType(MLIRContext *ctx, int64_t rank=ShapedType::kDynamic)
Alias type for extent tensors.
Include the generated interface declarations.