| /llvm-project/mlir/lib/Dialect/Tensor/Transforms/ |
| H A D | IndependenceTransforms.cpp | 114 SmallVector<OpFoldResult> newSizes; in buildIndependentOp() local
119 newSizes.push_back(*ub); in buildIndependentOp()
123 if (llvm::equal(emptyOp.getMixedSizes(), newSizes)) in buildIndependentOp()
128 b.create<EmptyOp>(loc, newSizes, emptyOp.getType().getElementType()); in buildIndependentOp()
131 SmallVector<OpFoldResult> offsets(newSizes.size(), b.getIndexAttr(0)); in buildIndependentOp()
132 SmallVector<OpFoldResult> strides(newSizes.size(), b.getIndexAttr(1)); in buildIndependentOp()
|
| H A D | MergeConsecutiveInsertExtractSlicePatterns.cpp | 32 SmallVector<OpFoldResult> newOffsets, newSizes, newStrides; in matchAndRewrite() local
35 newOffsets, newSizes, newStrides))) in matchAndRewrite()
40 newSizes, newStrides); in matchAndRewrite()
121 SmallVector<OpFoldResult> newOffsets, newSizes, newStrides; in matchAndRewrite() local
127 newSizes.push_back(extractSliceOp.getMixedSizes()[i]); in matchAndRewrite()
132 newSizes, newStrides); in matchAndRewrite()
|
| /llvm-project/mlir/lib/Dialect/MemRef/Transforms/ |
| H A D | IndependenceTransforms.cpp | 42 SmallVector<OpFoldResult> newSizes; in buildIndependentOp() local
47 newSizes.push_back(*ub); in buildIndependentOp()
51 if (llvm::equal(allocaOp.getMixedSizes(), newSizes)) in buildIndependentOp()
56 b.create<AllocaOp>(loc, newSizes, allocaOp.getType().getElementType()); in buildIndependentOp()
59 SmallVector<OpFoldResult> offsets(newSizes.size(), b.getIndexAttr(0)); in buildIndependentOp()
60 SmallVector<OpFoldResult> strides(newSizes.size(), b.getIndexAttr(1)); in buildIndependentOp()
|
| /llvm-project/mlir/include/mlir/Dialect/Tosa/Utils/ |
| H A D | ShapeUtils.h | 35 ValueKnowledge(bool hasRank, llvm::ArrayRef<int64_t> newSizes, Type dtype) in ValueKnowledge()
37 sizes.reserve(newSizes.size()); in ValueKnowledge()
38 for (auto size : newSizes) in ValueKnowledge()
|
| /llvm-project/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | VectorDropLeadUnitDim.cpp | 88 auto newSizes = rewriter.getArrayAttr( in matchAndRewrite() local
94 loc, newDstType, newSrcVector, newOffsets, newSizes, newStrides); in matchAndRewrite()
|
| H A D | VectorTransforms.cpp | 685 ArrayAttr newSizes = extractOp.getSizes(); in matchAndRewrite()
686 if (newSizes.size() == rank) { in matchAndRewrite() local
687 SmallVector<int64_t> sizes = getIntValueVector(newSizes); in matchAndRewrite()
691 newSizes = rewriter.getI64ArrayAttr(sizes); in matchAndRewrite()
702 newSizes, extractOp.getStrides()); in matchAndRewrite()
|
| /llvm-project/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgOps.cpp | 789 SmallVector<OpFoldResult, 4> newSizes;
792 newSizes.push_back( in matchAndRewrite()
796 newSizes.push_back(rewriter.getIndexAttr(srcPadType.getDimSize(i))); in matchAndRewrite()
802 newSizes, insertOp.getMixedStrides()); in matchAndRewrite()
768 SmallVector<OpFoldResult, 4> newSizes; matchAndRewrite() local
|
| /llvm-project/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorOps.cpp | 3404 SmallVector<OpFoldResult> newSizes = innerSliceOp.getMixedSizes();
3405 for (auto en : enumerate(newSizes)) {
3432 padOp.getLoc(), outerSliceOp.getSource(), newOffsets, newSizes,
3249 SmallVector<OpFoldResult> newSizes = innerSliceOp.getMixedSizes(); matchAndRewrite() local
|