Searched refs:mixedOffsets (Results 1 – 5 of 5) sorted by relevance
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Arithmetic/Utils/ |
| H A D | Utils.h | 54 SmallVector<OpFoldResult> mixedOffsets(op.getMixedOffsets()); in matchAndRewrite()
57 canonicalizeSubViewPart(mixedOffsets, ShapedType::isDynamicStrideOrOffset); in matchAndRewrite()
64 resultTypeFunc(op, mixedOffsets, mixedSizes, mixedStrides); in matchAndRewrite()
69 mixedOffsets, mixedSizes, mixedStrides); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/ |
| H A D | FoldSubViewOps.cpp | 47 SmallVector<OpFoldResult> mixedOffsets = subViewOp.getMixedOffsets(); in resolveSourceIndices() local
62 if (useIndices.size() != mixedOffsets.size()) in resolveSourceIndices()
65 for (auto index : llvm::seq<size_t>(0, mixedOffsets.size())) { in resolveSourceIndices()
80 if (auto attr = mixedOffsets[index].dyn_cast<Attribute>()) { in resolveSourceIndices()
83 dynamicOperands.push_back(mixedOffsets[index].get<Value>()); in resolveSourceIndices()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/Transforms/ |
| H A D | BufferizableOpInterfaceImpl.cpp | 281 SmallVector<OpFoldResult> mixedOffsets = extractSliceOp.getMixedOffsets(); in bufferize() local
296 extractSliceOp.getType().getShape(), srcMemrefType, mixedOffsets, in bufferize()
300 loc, subviewMemRefType, *srcMemref, mixedOffsets, mixedSizes, in bufferize()
706 SmallVector<OpFoldResult> mixedOffsets = insertSliceOp.getMixedOffsets(); 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/Tensor/IR/ |
| H A D | TensorOps.cpp | 1353 ArrayRef<OpFoldResult> mixedOffsets, in operator ()()
1357 op.getType().getRank(), op.getSourceType(), mixedOffsets, mixedSizes, in operator ()()
1607 SmallVector<OpFoldResult> mixedOffsets(insertSliceOp.getMixedOffsets()); in matchAndRewrite() local
1610 canonicalizeSubViewPart(mixedOffsets, ShapedType::isDynamicStrideOrOffset); in matchAndRewrite()
1617 mixedOffsets, mixedSizes, mixedStrides); in matchAndRewrite()
1630 insertSliceOp, toInsert, insertSliceOp.getDest(), mixedOffsets, in matchAndRewrite()
2313 SmallVector<OpFoldResult> mixedOffsets(insertSliceOp.getMixedOffsets()); in matchAndRewrite() local
2316 canonicalizeSubViewPart(mixedOffsets, ShapedType::isDynamicStrideOrOffset); in matchAndRewrite()
2324 insertSliceOp.getDestType(), mixedOffsets, mixedSizes, in matchAndRewrite()
2334 insertSliceOp, toInsert, insertSliceOp.getDest(), mixedOffsets, in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/ |
| H A D | MemRefOps.cpp | 2439 MemRefType sourceType, ArrayRef<OpFoldResult> mixedOffsets, in getCanonicalSubViewResultType() argument
2441 auto nonRankReducedType = SubViewOp::inferResultType(sourceType, mixedOffsets, in getCanonicalSubViewResultType()
2468 ArrayRef<OpFoldResult> mixedOffsets, ArrayRef<OpFoldResult> mixedSizes, in getCanonicalSubViewResultType() argument
2471 sourceType, mixedOffsets, mixedSizes, in getCanonicalSubViewResultType()
2483 auto mixedOffsets = subViewOp.getMixedOffsets(); in isTrivialSubViewOp() local
2488 if (llvm::any_of(mixedOffsets, [](OpFoldResult ofr) { in isTrivialSubViewOp()
2595 MemRefType operator()(SubViewOp op, ArrayRef<OpFoldResult> mixedOffsets, in operator ()()
2599 mixedOffsets, mixedSizes, in operator ()()
|