| /llvm-project-15.0.7/mlir/lib/Interfaces/ |
| H A D | ViewLikeInterface.cpp | 44 if (!(op.getMixedOffsets().size() == 1 && maxRanks[0] == 1) && // NOLINT in verifyOffsetSizeAndStrideOp()
45 op.getMixedOffsets().size() != op.getMixedSizes().size()) in verifyOffsetSizeAndStrideOp()
48 << op.getMixedOffsets().size() << " vs " << op.getMixedSizes().size() in verifyOffsetSizeAndStrideOp()
170 for (auto it : llvm::zip(a.getMixedOffsets(), b.getMixedOffsets())) in sameOffsetsSizesAndStrides()
183 mlir::getMixedOffsets(OffsetSizeAndStrideOpInterface op, in getMixedOffsets() function in mlir
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/ |
| H A D | ComposeSubView.cpp | 77 for (auto it : llvm::zip(op.getMixedOffsets(), sourceOp.getMixedOffsets(), in matchAndRewrite()
|
| H A D | MultiBuffer.cpp | 53 subviewUse.getMixedOffsets(), subviewUse.getMixedSizes(), in replaceUsesAndPropagateType()
|
| H A D | FoldSubViewOps.cpp | 47 SmallVector<OpFoldResult> mixedOffsets = subViewOp.getMixedOffsets(); in resolveSourceIndices()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/Transforms/ |
| H A D | SwapExtractSliceWithProducer.cpp | 37 producerOp.getDestinationOperands(builder), sliceOp.getMixedOffsets(), in replaceExtractSliceWithTiledProducer()
|
| H A D | BufferizableOpInterfaceImpl.cpp | 281 SmallVector<OpFoldResult> mixedOffsets = extractSliceOp.getMixedOffsets(); in bufferize()
706 SmallVector<OpFoldResult> mixedOffsets = insertSliceOp.getMixedOffsets(); in bufferize()
929 parallelInsertSliceOp.getMixedOffsets(), in bufferize()
935 parallelInsertSliceOp.getMixedOffsets(), in bufferize()
|
| /llvm-project-15.0.7/mlir/include/mlir/Interfaces/ |
| H A D | ViewLikeInterface.h | 36 SmallVector<OpFoldResult, 4> getMixedOffsets(OffsetSizeAndStrideOpInterface op,
|
| H A D | ViewLikeInterface.td | 164 /*methodName=*/"getMixedOffsets",
168 return ::mlir::getMixedOffsets($_op, $_op.static_offsets(),
433 return ::llvm::all_of(getMixedOffsets(), [](::mlir::OpFoldResult ofr) {
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | DropUnitDims.cpp | 467 SmallVector<OpFoldResult> offsets = sliceOp.getMixedOffsets(); in matchAndRewrite()
498 SmallVector<OpFoldResult> offsets = insertSliceOp.getMixedOffsets(); in matchAndRewrite()
519 insertSliceOp.getMixedOffsets(), insertSliceOp.getMixedSizes(), in matchAndRewrite()
|
| H A D | BubbleUpExtractSlice.cpp | 103 rewriter, sliceLoc, sliceOp.getMixedOffsets()); in matchAndRewrite()
|
| H A D | Hoisting.cpp | 85 for (auto it : llvm::zip(s.getMixedOffsets(), si.getMixedOffsets())) in sameOffsetsSizesAndStrides()
|
| H A D | TilingInterfaceImpl.cpp | 131 resultOffsets = sliceOp.getMixedOffsets(); in getResultTilePosition()
|
| H A D | Tiling.cpp | 181 loc, source, dest, subsetExtractOp.getMixedOffsets(), in createMatchingParallelSubsetInsertOp()
|
| H A D | Transforms.cpp | 832 tensor::bubbleUpPadSlice(rewriter, padOp, sliceOp.getMixedOffsets(), in matchAndRewrite()
|
| H A D | Vectorization.cpp | 1073 ofrToIndexValues(rewriter, padOp.getLoc(), insertOp.getMixedOffsets()); in rewriteUser()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Arithmetic/Utils/ |
| H A D | Utils.h | 54 SmallVector<OpFoldResult> mixedOffsets(op.getMixedOffsets()); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorOps.cpp | 275 extractOperand.getSource(), extractOperand.getMixedOffsets(), sizes, in matchAndRewrite()
1127 getSourceType(), getMixedOffsets(), getMixedSizes(), getMixedStrides()); in verify()
1211 sliceOp.getMixedOffsets(), sliceOp.getMixedSizes(), in matchAndRewrite()
1387 for (OpFoldResult ofr : op.getMixedOffsets()) in foldIdentityOffsetSizeAndStrideOpInterface()
1607 SmallVector<OpFoldResult> mixedOffsets(insertSliceOp.getMixedOffsets()); in matchAndRewrite()
1691 insertSliceOp.getLoc(), src, dst, insertSliceOp.getMixedOffsets(), in matchAndRewrite()
1768 insertSliceOp.getMixedOffsets(), insertSliceOp.getMixedSizes(), in matchAndRewrite()
2132 OpFoldResult innerOffset = innerSliceOp.getMixedOffsets()[en.index()]; in matchAndRewrite()
2133 OpFoldResult outerOffset = outerSliceOp.getMixedOffsets()[en.index()]; in matchAndRewrite()
2313 SmallVector<OpFoldResult> mixedOffsets(insertSliceOp.getMixedOffsets()); 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/Vector/IR/ |
| H A D | VectorOps.cpp | 3255 extractOp.getSourceType(), extractOp.getMixedOffsets(), in matchAndRewrite()
3267 OpFoldResult offset = extractOp.getMixedOffsets()[i]; in matchAndRewrite()
3273 extractOp.getMixedOffsets()[it.index() + rankReduced]; in matchAndRewrite()
3757 insertOp.getType(), insertOp.getMixedOffsets(), in matchAndRewrite()
3766 rewriter, insertOp.getLoc(), insertOp.getMixedOffsets()); in matchAndRewrite()
3864 insertOp.getMixedOffsets(), insertOp.getMixedSizes(), in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/ |
| H A D | MemRefOps.cpp | 1587 if (llvm::all_of(prev.getMixedOffsets(), [](OpFoldResult val) { in fold()
2483 auto mixedOffsets = subViewOp.getMixedOffsets(); in isTrivialSubViewOp()
2556 subViewOp.getMixedOffsets(), subViewOp.getMixedSizes(), in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/TransformOps/ |
| H A D | LinalgTransformOps.cpp | 257 sliceOp.getMixedOffsets(), sliceOp.getMixedSizes(), true); in tileAndFuse()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/ |
| H A D | Utils.cpp | 365 for (const auto &en : enumerate(producerOp.getMixedOffsets())) { in makeComposedExtractSliceOp()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgOps.cpp | 620 SmallVector<OpFoldResult> oldOffsets = insertOp.getMixedOffsets(); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/MemRefToLLVM/ |
| H A D | MemRefToLLVM.cpp | 1717 subViewOp.getMixedOffsets().size()); in matchAndRewrite()
|