Home
last modified time | relevance | path

Searched refs:newOffsets (Results 1 – 5 of 5) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/
H A DTensorTilingInterfaceImpl.cpp125 SmallVector<OpFoldResult> newOffsets, newLengths, newStrides; in bubbleUpPadSlice() local
168 newOffsets.push_back(getAsOpFoldResult(newOffset)); in bubbleUpPadSlice()
249 loc, padOp.getSource(), newOffsets, newLengths, newStrides); in bubbleUpPadSlice()
H A DTensorOps.cpp2130 SmallVector<OpFoldResult> newOffsets(rank, rewriter.getIndexAttr(0)); in matchAndRewrite() local
2131 for (auto &en : enumerate(newOffsets)) { in matchAndRewrite()
2181 padOp.getLoc(), outerSliceOp.getSource(), newOffsets, newSizes, in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorDropLeadUnitDim.cpp71 auto newOffsets = rewriter.getArrayAttr( in matchAndRewrite() local
79 loc, newDstType, newSrcVector, newOffsets, newSizes, newStrides); in matchAndRewrite()
114 auto newOffsets = rewriter.getArrayAttr( in matchAndRewrite() local
120 loc, newDstType, newSrcVector, newDstVector, newOffsets, newStrides); in matchAndRewrite()
H A DVectorTransforms.cpp2354 ArrayAttr newOffsets = extractOp.getOffsets(); in matchAndRewrite() local
2355 if (newOffsets.size() == rank) { in matchAndRewrite()
2356 SmallVector<int64_t, 4> offsets = getIntValueVector(newOffsets); in matchAndRewrite()
2360 newOffsets = rewriter.getI64ArrayAttr(offsets); in matchAndRewrite()
2380 extractOp.getLoc(), newExtractType, castOp.getSource(), newOffsets, in matchAndRewrite()
2435 ArrayAttr newOffsets = insertOp.getOffsets(); in matchAndRewrite() local
2436 assert(newOffsets.size() == rank); in matchAndRewrite()
2437 SmallVector<int64_t, 4> offsets = getIntValueVector(newOffsets); in matchAndRewrite()
2441 newOffsets = rewriter.getI64ArrayAttr(offsets); in matchAndRewrite()
2462 bitcastOp, bitcastOp.getType(), newCastSrcOp, newCastDstOp, newOffsets, in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/
H A DLinalgOps.cpp631 SmallVector<OpFoldResult, 4> newOffsets; in matchAndRewrite() local
637 newOffsets.push_back( in matchAndRewrite()
649 insertOp, srcPadOp.getSource(), insertOp.getDest(), newOffsets, in matchAndRewrite()