| /llvm-project-15.0.7/mlir/lib/Interfaces/ |
| H A D | ViewLikeInterface.cpp | 45 op.getMixedOffsets().size() != op.getMixedSizes().size()) in verifyOffsetSizeAndStrideOp()
48 << op.getMixedOffsets().size() << " vs " << op.getMixedSizes().size() in verifyOffsetSizeAndStrideOp()
52 if (op.getMixedSizes().size() != op.getMixedStrides().size()) in verifyOffsetSizeAndStrideOp()
55 << op.getMixedSizes().size() << " vs " << op.getMixedStrides().size() in verifyOffsetSizeAndStrideOp()
173 for (auto it : llvm::zip(a.getMixedSizes(), b.getMixedSizes())) in sameOffsetsSizesAndStrides()
198 mlir::getMixedSizes(OffsetSizeAndStrideOpInterface op, ArrayAttr staticSizes, in getMixedSizes() function in mlir
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/Transforms/ |
| H A D | SwapExtractSliceWithProducer.cpp | 38 sliceOp.getMixedSizes(), true); in replaceExtractSliceWithTiledProducer()
|
| H A D | BufferizableOpInterfaceImpl.cpp | 282 SmallVector<OpFoldResult> mixedSizes = extractSliceOp.getMixedSizes(); in bufferize()
707 SmallVector<OpFoldResult> mixedSizes = insertSliceOp.getMixedSizes(); in bufferize()
930 parallelInsertSliceOp.getMixedSizes(), in bufferize()
936 parallelInsertSliceOp.getMixedSizes(), in bufferize()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorOps.cpp | 259 SmallVector<OpFoldResult, 4> sizes = extractOperand.getMixedSizes(); in matchAndRewrite()
1134 SmallVector<OpFoldResult> mixedSizes = getMixedSizes(); in getDroppedDims()
1156 SmallVector<OpFoldResult> mixedSizes = getMixedSizes(); in reifyResultShapes()
1211 sliceOp.getMixedOffsets(), sliceOp.getMixedSizes(), in matchAndRewrite()
1393 for (auto it : llvm::zip(op.getMixedSizes(), shape)) in foldIdentityOffsetSizeAndStrideOpInterface()
1608 SmallVector<OpFoldResult> mixedSizes(insertSliceOp.getMixedSizes()); in matchAndRewrite()
1692 insertSliceOp.getMixedSizes(), insertSliceOp.getMixedStrides()); in matchAndRewrite()
1742 getConstantIntValue(insertSliceOp.getMixedSizes()[i])) in matchAndRewrite()
1768 insertSliceOp.getMixedOffsets(), insertSliceOp.getMixedSizes(), in matchAndRewrite()
2153 SmallVector<OpFoldResult> newSizes = innerSliceOp.getMixedSizes(); in matchAndRewrite()
[all …]
|
| /llvm-project-15.0.7/mlir/include/mlir/Interfaces/ |
| H A D | ViewLikeInterface.h | 42 SmallVector<OpFoldResult, 4> getMixedSizes(OffsetSizeAndStrideOpInterface op,
|
| H A D | ViewLikeInterface.td | 177 /*methodName=*/"getMixedSizes",
181 return ::mlir::getMixedSizes($_op, $_op.static_sizes(), $_op.sizes());
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/ |
| H A D | ComposeSubView.cpp | 78 op.getMixedSizes())) { in matchAndRewrite()
|
| H A D | MultiBuffer.cpp | 53 subviewUse.getMixedOffsets(), subviewUse.getMixedSizes(), in replaceUsesAndPropagateType()
|
| H A D | FoldSubViewOps.cpp | 48 SmallVector<OpFoldResult> mixedSizes = subViewOp.getMixedSizes(); in resolveSourceIndices()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | DropUnitDims.cpp | 468 SmallVector<OpFoldResult> sizes = sliceOp.getMixedSizes(); in matchAndRewrite()
499 SmallVector<OpFoldResult> sizes = insertSliceOp.getMixedSizes(); in matchAndRewrite()
519 insertSliceOp.getMixedOffsets(), insertSliceOp.getMixedSizes(), in matchAndRewrite()
|
| H A D | BubbleUpExtractSlice.cpp | 105 sliceOp.getMixedSizes()); in matchAndRewrite()
|
| H A D | Hoisting.cpp | 88 for (auto it : llvm::zip(s.getMixedSizes(), si.getMixedSizes())) in sameOffsetsSizesAndStrides()
|
| H A D | Vectorization.cpp | 959 beforeSlice.getMixedSizes().size()); in hasSameTensorSize()
961 afterTrimming.getMixedSizes().size()); in hasSameTensorSize()
967 auto size1 = beforeSlice.getMixedSizes()[i]; in hasSameTensorSize()
968 auto size2 = afterTrimming.getMixedSizes()[i]; in hasSameTensorSize()
1053 llvm::zip(insertOp.getMixedSizes(), expectedSizes), [](auto it) { in rewriteUser()
|
| H A D | TilingInterfaceImpl.cpp | 132 resultSizes = sliceOp.getMixedSizes(); in getResultTilePosition()
|
| H A D | Transforms.cpp | 219 for (const auto &en : enumerate(shapedOp.getMixedSizes())) { in padOperandToSmallestStaticBoundingBox()
833 sliceOp.getMixedSizes(), zeroSliceGuard); in matchAndRewrite()
|
| H A D | Tiling.cpp | 182 subsetExtractOp.getMixedSizes(), subsetExtractOp.getMixedStrides()); in createMatchingParallelSubsetInsertOp()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Arithmetic/Utils/ |
| H A D | Utils.h | 55 SmallVector<OpFoldResult> mixedSizes(op.getMixedSizes()); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/Transforms/ |
| H A D | AllocTensorElimination.cpp | 220 insertOp.getMixedOffsets(), insertOp.getMixedSizes(), in insertSliceAnchoredAllocTensorEliminationStep()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/ |
| H A D | Utils.cpp | 413 sliceOp.getMixedSizes().size() != padOpSliceOp.getMixedSizes().size()) in makeComposedPadHighOp()
419 llvm::zip(sliceOp.getMixedSizes(), padOpSliceOp.getMixedSizes()), in makeComposedPadHighOp()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/TransformOps/ |
| H A D | LinalgTransformOps.cpp | 257 sliceOp.getMixedOffsets(), sliceOp.getMixedSizes(), true); in tileAndFuse()
910 for (OpFoldResult ofr : getMixedSizes()) { in apply()
943 SmallVector<OpFoldResult> transform::TileOp::getMixedSizes() { in getMixedSizes() function in transform::TileOp
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/IR/ |
| H A D | LinalgOps.td | 94 SmallVector<OpFoldResult> getMixedSizes();
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/ |
| H A D | MemRefOps.cpp | 868 computeMemRefRankReductionMask(sourceType, resultType, getMixedSizes()); in getDroppedDims()
2392 subViewType, getMixedSizes()); in verify()
2484 auto mixedSizes = subViewOp.getMixedSizes(); in isTrivialSubViewOp()
2556 subViewOp.getMixedOffsets(), subViewOp.getMixedSizes(), in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgOps.cpp | 1230 SmallVector<OpFoldResult> InitTensorOp::getMixedSizes() { in getMixedSizes() function in InitTensorOp
1398 SmallVector<OpFoldResult> currMixedSizes = producer.getMixedSizes(); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/ |
| H A D | VectorOps.cpp | 3256 extractOp.getMixedSizes(), extractOp.getMixedStrides()); in matchAndRewrite()
3758 insertOp.getMixedSizes(), insertOp.getMixedStrides()); in matchAndRewrite()
3838 llvm::zip(insertOp.getMixedSizes(), extractOp.getMixedSizes())) { in matchAndRewrite()
3864 insertOp.getMixedOffsets(), insertOp.getMixedSizes(), in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/TransformOps/ |
| H A D | LinalgTransformOps.td | 657 SmallVector<OpFoldResult> getMixedSizes();
|