Home
last modified time | relevance | path

Searched refs:createOrFold (Results 1 – 25 of 32) sorted by relevance

12

/llvm-project-15.0.7/mlir/include/mlir/IR/
H A DImplicitLocOpBuilder.h61 using OpBuilder::createOrFold;
74 void createOrFold(llvm::SmallVectorImpl<Value> &results, Args &&...args) { in createOrFold() function
75 OpBuilder::createOrFold<OpTy>(results, curLoc, std::forward<Args>(args)...); in createOrFold()
82 createOrFold(Args &&...args) { in createOrFold() function
83 return OpBuilder::createOrFold<OpTy>(curLoc, std::forward<Args>(args)...); in createOrFold()
90 createOrFold(Args &&...args) { in createOrFold() function
91 return OpBuilder::createOrFold<OpTy>(curLoc, std::forward<Args>(args)...); in createOrFold()
H A DBuilders.h458 void createOrFold(SmallVectorImpl<Value> &results, Location location, in createOrFold() function
478 createOrFold(Location location, Args &&...args) { in createOrFold() function
480 createOrFold<OpTy>(results, location, std::forward<Args>(args)...); in createOrFold()
488 createOrFold(Location location, Args &&...args) { in createOrFold() function
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DPromotion.cpp60 return b.createOrFold<memref::AllocaOp>(staticBufferType, ValueRange{}, in allocBuffer()
63 return b.createOrFold<memref::AllocOp>(staticBufferType, ValueRange{}, in allocBuffer()
69 Value mul = b.createOrFold<arith::MulIOp>( in allocBuffer()
87 auto zero = b.createOrFold<arith::ConstantIndexOp>(0); in defaultAllocBufferCallBack()
88 auto one = b.createOrFold<arith::ConstantIndexOp>(1); in defaultAllocBufferCallBack()
92 allocSize = b.createOrFold<arith::MulIOp>(allocSize, size.value()); in defaultAllocBufferCallBack()
97 Value view = b.createOrFold<memref::ViewOp>( in defaultAllocBufferCallBack()
237 b.createOrFold<memref::DimOp>(loc, subView, resultDimIdx++)); in promoteSubviewAsNewBuffer()
247 auto partialLocalView = b.createOrFold<memref::SubViewOp>( in promoteSubviewAsNewBuffer()
H A DHoistPadding.cpp356 Value ubVal = b.createOrFold<AffineMinOp>(boundMap, boundOperands); in getPackedTensorSizes()
364 Value res = b.createOrFold<AffineApplyOp>( in getPackedTensorSizes()
393 return b.createOrFold<AffineApplyOp>(loc, (iv - lb).ceilDiv(step), in buildLoopIterationCount()
H A DTransforms.cpp770 auto srcSize = rewriter.createOrFold<tensor::DimOp>( in matchAndRewrite()
773 auto plusLow = rewriter.createOrFold<arith::AddIOp>( in matchAndRewrite()
775 auto plusHigh = rewriter.createOrFold<arith::AddIOp>( in matchAndRewrite()
798 srcSizes.push_back(rewriter.createOrFold<tensor::DimOp>( in matchAndRewrite()
H A DTilingInterfaceImpl.cpp113 return builder.createOrFold<AffineApplyOp>(loc, map, normalizedOperands); in getResultTilePosition()
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/
H A DTensorTilingInterfaceImpl.cpp93 return b.createOrFold<AffineApplyOp>(loc, addMap, ValueRange{v1, v2}); in bubbleUpPadSlice()
98 return b.createOrFold<AffineApplyOp>(loc, subMap, ValueRange{v1, v2}); in bubbleUpPadSlice()
103 return b.createOrFold<AffineMinOp>(loc, idMap, ValueRange{v1, v2}); in bubbleUpPadSlice()
107 return b.createOrFold<AffineMaxOp>(loc, idMap, ValueRange{v1, v2}); in bubbleUpPadSlice()
144 auto srcSize = b.createOrFold<tensor::DimOp>(loc, padOp.getSource(), dim); in bubbleUpPadSlice()
H A DTensorInferTypeOpInterfaceImpl.cpp51 dynamicDims.push_back(builder.createOrFold<tensor::DimOp>(loc, src, dim)); in getCollapsedOutputDimFromInputShape()
181 b.createOrFold<tensor::DimOp>(loc, padOp.getSource(), dim)); in reifyResultShapes()
H A DTensorOps.cpp1442 dim = b.createOrFold<tensor::DimOp>( in createCanonicalRankReducingExtractSliceOp()
1449 return b.createOrFold<tensor::ExtractSliceOp>(loc, targetType, tensor, in createCanonicalRankReducingExtractSliceOp()
1581 builder.createOrFold<tensor::DimOp>(getLoc(), getDest(), dim); in reifyResultShapes()
1795 dim = b.createOrFold<tensor::DimOp>( in createCanonicalRankReducingInsertSliceOp()
1802 return b.createOrFold<tensor::InsertSliceOp>(loc, tensor, dest, offsets, in createCanonicalRankReducingInsertSliceOp()
/llvm-project-15.0.7/mlir/lib/Conversion/BufferizationToMemRef/
H A DBufferizationToMemRef.cpp59 Value size = rewriter.createOrFold<arith::ConstantIndexOp>(loc, i); in matchAndRewrite()
61 rewriter.createOrFold<memref::DimOp>(loc, op.getInput(), size); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/Utils/
H A DUtils.cpp42 auto zero = b.createOrFold<arith::ConstantIndexOp>(loc, 0); in createPadHighOp()
52 auto dimOp = b.createOrFold<tensor::DimOp>(loc, source, en.index()); in createPadHighOp()
/llvm-project-15.0.7/mlir/lib/Dialect/NVGPU/Transforms/
H A DOptimizeSharedMemory.cpp79 srcBits = b.createOrFold<arith::ShLIOp>(loc, srcBits, finalShiftVal); in permuteVectorOffset()
82 srcBits = b.createOrFold<arith::ShRUIOp>(loc, srcBits, finalShiftVal); in permuteVectorOffset()
86 srcBits = b.createOrFold<arith::ShLIOp>(loc, srcBits, finalShiftVal); in permuteVectorOffset()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Utils/
H A DVectorUtils.cpp40 return b.createOrFold<memref::DimOp>(loc, source, dim); in createOrFoldDimOp()
42 return b.createOrFold<tensor::DimOp>(loc, source, dim); in createOrFoldDimOp()
/llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/
H A DTosaToLinalg.cpp1652 Value axisValue = rewriter.createOrFold<arith::ConstantOp>( in matchAndRewrite()
1662 sizes.push_back(rewriter.createOrFold<tensor::DimOp>( in matchAndRewrite()
1672 auto size = rewriter.createOrFold<tensor::DimOp>(loc, arg, axisValue); in matchAndRewrite()
1674 rewriter.createOrFold<arith::AddIOp>(loc, resultDimSize, size); in matchAndRewrite()
1681 Value zeroVal = rewriter.createOrFold<arith::ConstantOp>( in matchAndRewrite()
1696 result = rewriter.createOrFold<tensor::InsertSliceOp>( in matchAndRewrite()
1852 padConstant = rewriter.createOrFold<tensor::ExtractOp>( in matchAndRewrite()
1886 Value lowVal = rewriter.createOrFold<tensor::ExtractOp>( in matchAndRewrite()
1888 Value highVal = rewriter.createOrFold<tensor::ExtractOp>( in matchAndRewrite()
1891 lowVal = rewriter.createOrFold<arith::IndexCastOp>( in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/
H A DResolveShapedTypeResultDims.cpp61 rewriter.createOrFold<arith::ConstantIndexOp>(loc, *dimIndex)); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/GPU/Transforms/
H A DMemoryPromotion.cpp54 ubs.push_back(b.createOrFold<memref::DimOp>( in insertCopyLoops()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Transforms/
H A DPipelineDataTransfer.cpp89 allocOperands.push_back(bOuter.createOrFold<memref::DimOp>( in doubleBuffer()
/llvm-project-15.0.7/mlir/lib/Dialect/SCF/Transforms/
H A DLoopSpecialization.cpp131 splitBound = b.createOrFold<AffineApplyOp>(loc, modMap, in peelForLoop()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp5087 return b.createOrFold<arith::AddIOp>(loc, v1, v2); in makeArithReduction()
5089 return b.createOrFold<arith::AddFOp>(loc, v1, v2); in makeArithReduction()
5093 return b.createOrFold<arith::AndIOp>(loc, v1, v2); in makeArithReduction()
5097 return b.createOrFold<arith::MaxFOp>(loc, v1, v2); in makeArithReduction()
5101 return b.createOrFold<arith::MinFOp>(loc, v1, v2); in makeArithReduction()
5104 return b.createOrFold<arith::MaxSIOp>(loc, v1, v2); in makeArithReduction()
5107 return b.createOrFold<arith::MinSIOp>(loc, v1, v2); in makeArithReduction()
5110 return b.createOrFold<arith::MaxUIOp>(loc, v1, v2); in makeArithReduction()
5113 return b.createOrFold<arith::MinUIOp>(loc, v1, v2); in makeArithReduction()
5122 return b.createOrFold<arith::OrIOp>(loc, v1, v2); in makeArithReduction()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/
H A DLinalgInterfaces.cpp484 return b.createOrFold<memref::DimOp>(loc, source, dim); in createOrFoldDimOp()
486 return b.createOrFold<tensor::DimOp>(loc, source, dim); in createOrFoldDimOp()
/llvm-project-15.0.7/mlir/test/lib/Dialect/Test/
H A DTestDialect.cpp1182 builder.createOrFold<tensor::DimOp>(getLoc(), operands.front(), 0)}; in reifyReturnTypeShapes()
1195 return builder.createOrFold<tensor::DimOp>(loc, operand, dim); in reifyReturnTypeShapes()
1213 return builder.createOrFold<tensor::DimOp>(loc, operand, dim); in reifyResultShapes()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/
H A DUtils.cpp197 return b.createOrFold<memref::DimOp>(loc, source, dim); in createOrFoldDimOp()
199 return b.createOrFold<tensor::DimOp>(loc, source, dim); in createOrFoldDimOp()
796 return b.createOrFold<AffineApplyOp>(loc, map, normalizedOperands); in fullyComposeAndAffineApply()
/llvm-project-15.0.7/mlir/lib/Dialect/Async/Transforms/
H A DAsyncParallelFor.cpp740 auto range = b.createOrFold<arith::SubIOp>(ub, lb); in matchAndRewrite()
741 tripCounts[i] = b.createOrFold<arith::CeilDivSIOp>(range, step); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Conversion/NVGPUToNVVM/
H A DNVGPUToNVVM.cpp91 el = rewriter.createOrFold<LLVM::BitcastOp>( in convertIntrinsicResult()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/IR/
H A DAffineOps.cpp745 createOrFold(RewriterBase &b, Location loc, ValueRange operands, in createOrFold() function
803 OpFoldResult result = createOrFold<AffineApplyOp>(b, loc, actualValues, map); in makeComposedFoldedAffineApply()
854 return b.createOrFold<AffineMinOp>(loc, b.getIndexType(), map, allOperands); in makeComposedAffineMin()
865 createOrFold<AffineMinOp>(b, loc, actualValues, b.getIndexType(), map); in makeComposedFoldedAffineMin()
881 return b.createOrFold<AffineApplyOp>(loc, map, operands); in createFoldedComposedAffineApply()
918 results.push_back(createOrFold<AffineApplyOp>(b, loc, operands, submap)); in applyMapToValues()

12