| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Arithmetic/Utils/ |
| H A D | Utils.h | 55 SmallVector<OpFoldResult> mixedSizes(op.getMixedSizes()); in matchAndRewrite()
58 canonicalizeSubViewPart(mixedSizes, ShapedType::isDynamic); in matchAndRewrite()
64 resultTypeFunc(op, mixedOffsets, mixedSizes, mixedStrides); in matchAndRewrite()
69 mixedOffsets, mixedSizes, mixedStrides); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorOps.cpp | 1134 SmallVector<OpFoldResult> mixedSizes = getMixedSizes(); in getDroppedDims() local
1135 llvm::SmallBitVector droppedDims(mixedSizes.size()); in getDroppedDims()
1137 for (const auto &size : enumerate(mixedSizes)) { in getDroppedDims()
1156 SmallVector<OpFoldResult> mixedSizes = getMixedSizes(); in reifyResultShapes() local
1159 for (const auto &size : enumerate(mixedSizes)) { in reifyResultShapes()
1354 ArrayRef<OpFoldResult> mixedSizes, in operator ()()
1611 canonicalizeSubViewPart(mixedSizes, ShapedType::isDynamic); in matchAndRewrite()
1617 mixedOffsets, mixedSizes, mixedStrides); in matchAndRewrite()
1631 mixedSizes, mixedStrides); in matchAndRewrite()
2317 canonicalizeSubViewPart(mixedSizes, ShapedType::isDynamic); in matchAndRewrite()
[all …]
|
| H A D | TensorTilingInterfaceImpl.cpp | 31 SmallVector<OpFoldResult> mixedSizes = getAsOpFoldResult(reifiedShapes[0]); in getDestinationOperands() local
33 op->getLoc(), mixedSizes, padOp.getResultType().getElementType()); in getDestinationOperands()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/Transforms/ |
| H A D | BufferizableOpInterfaceImpl.cpp | 282 SmallVector<OpFoldResult> mixedSizes = extractSliceOp.getMixedSizes(); in bufferize() local
297 mixedSizes, mixedStrides) in bufferize()
300 loc, subviewMemRefType, *srcMemref, mixedOffsets, mixedSizes, in bufferize()
707 SmallVector<OpFoldResult> mixedSizes = insertSliceOp.getMixedSizes(); in bufferize() local
722 mixedOffsets, mixedSizes, mixedStrides) in bufferize()
725 loc, subviewMemRefType, *dstMemref, mixedOffsets, mixedSizes, in bufferize()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/ |
| H A D | MemRefOps.cpp | 2440 ArrayRef<OpFoldResult> mixedSizes, ArrayRef<OpFoldResult> mixedStrides) { in getCanonicalSubViewResultType() argument
2442 mixedSizes, mixedStrides) in getCanonicalSubViewResultType()
2446 mixedSizes); in getCanonicalSubViewResultType()
2468 ArrayRef<OpFoldResult> mixedOffsets, ArrayRef<OpFoldResult> mixedSizes, in getCanonicalSubViewResultType() argument
2471 sourceType, mixedOffsets, mixedSizes, in getCanonicalSubViewResultType()
2484 auto mixedSizes = subViewOp.getMixedSizes(); in isTrivialSubViewOp() local
2503 for (const auto &size : llvm::enumerate(mixedSizes)) { in isTrivialSubViewOp()
2596 ArrayRef<OpFoldResult> mixedSizes, in operator ()()
2599 mixedOffsets, mixedSizes, in operator ()()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Utils/ |
| H A D | Utils.h | 153 getReassociationMapForFoldingUnitDims(ArrayRef<OpFoldResult> mixedSizes);
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/ |
| H A D | FoldSubViewOps.cpp | 48 SmallVector<OpFoldResult> mixedSizes = subViewOp.getMixedSizes(); in resolveSourceIndices() local
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgOps.cpp | 1231 SmallVector<OpFoldResult> mixedSizes; in getMixedSizes() local
1232 mixedSizes.reserve(getType().getRank()); in getMixedSizes()
1237 mixedSizes.push_back(intAttr); in getMixedSizes()
1240 mixedSizes.push_back(sizes()[dynamicValIndex++]); in getMixedSizes()
1242 return mixedSizes; in getMixedSizes()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/ |
| H A D | Utils.cpp | 1084 getReassociationMapForFoldingUnitDims(ArrayRef<OpFoldResult> mixedSizes) { in getReassociationMapForFoldingUnitDims() argument
1087 for (const auto &it : llvm::enumerate(mixedSizes)) { in getReassociationMapForFoldingUnitDims()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/MemRefToLLVM/ |
| H A D | MemRefToLLVM.cpp | 1733 SmallVector<OpFoldResult> mixedSizes = subViewOp.getMixedSizes(); in matchAndRewrite() local
1735 assert(mixedSizes.size() == mixedStrides.size() && in matchAndRewrite()
1747 if (static_cast<unsigned>(i) >= mixedSizes.size()) { in matchAndRewrite()
|