|
std::optional< llvm::SmallDenseSet< unsigned > > | mlir::computeRankReductionMask (ArrayRef< int64_t > originalShape, ArrayRef< int64_t > reducedShape, bool matchDynamic=false) |
| Given an originalShape and a reducedShape assumed to be a subset of originalShape with some 1 entries erased, return the set of indices that specifies which of the entries of originalShape are dropped to obtain reducedShape . More...
|
|
SliceVerificationResult | mlir::isRankReducedType (ShapedType originalType, ShapedType candidateReducedType) |
| Check if originalType can be rank reduced to candidateReducedType type by dropping some dimensions with static size 1 . More...
|
|
LogicalResult | mlir::getStridesAndOffset (MemRefType t, SmallVectorImpl< int64_t > &strides, int64_t &offset) |
| Returns the strides of the MemRef if the layout map is in strided form. More...
|
|
std::pair< SmallVector< int64_t >, int64_t > | mlir::getStridesAndOffset (MemRefType t) |
| Wrapper around getStridesAndOffset(MemRefType, SmallVectorImpl<int64_t>, int64_t) that will assert if the logical result is not succeeded. More...
|
|
MemRefType | mlir::canonicalizeStridedLayout (MemRefType t) |
| Return a version of t with identity layout if it can be determined statically that the layout is the canonical contiguous strided layout. More...
|
|
AffineExpr | mlir::makeCanonicalStridedLayoutExpr (ArrayRef< int64_t > sizes, ArrayRef< AffineExpr > exprs, MLIRContext *context) |
| Given MemRef sizes that are either static or dynamic, returns the canonical "contiguous" strides AffineExpr. More...
|
|
AffineExpr | mlir::makeCanonicalStridedLayoutExpr (ArrayRef< int64_t > sizes, MLIRContext *context) |
| Return the result of makeCanonicalStrudedLayoutExpr for the common case where exprs is {d0, d1, .., d_(sizes.size()-1)}. More...
|
|
bool | mlir::isStrided (MemRefType t) |
| Return "true" if the layout for t is compatible with strided semantics. More...
|
|
bool | mlir::isLastMemrefDimUnitStride (MemRefType type) |
| Return "true" if the last dimension of the given type has a static unit stride. More...
|
|
bool | mlir::trailingNDimsContiguous (MemRefType type, int64_t n) |
| Return "true" if the last N dimensions of the given type are contiguous. More...
|
|