Home
last modified time | relevance | path

Searched defs:yieldOp (Results 1 – 25 of 42) sorted by relevance

12

/llvm-project/mlir/lib/Dialect/Shape/Transforms/
H A DBufferizableOpInterfaceImpl.cpp43 auto yieldOp = dyn_cast<shape::AssumingYieldOp>( in getAliasingOpOperands() local
54 auto yieldOp = cast<shape::AssumingYieldOp>( in bufferize() local
116 auto yieldOp = cast<shape::AssumingYieldOp>(op); in bufferize() local
/llvm-project/mlir/lib/Dialect/SCF/Transforms/
H A DBufferizableOpInterfaceImpl.cpp131 if (auto yieldOp = dyn_cast<scf::YieldOp>(block.getTerminator())) { in getUniqueYieldOp() local
176 auto yieldOp = getUniqueYieldOp(executeRegionOp); in getAliasingOpOperands() local
186 auto yieldOp = getUniqueYieldOp(executeRegionOp); in bufferize() local
334 auto yieldOp = getAliasingOpOperands() local
396 auto yieldOp = cast<scf::YieldOp>(b.getTerminator()); getBufferType() local
661 auto yieldOp = cast<scf::YieldOp>(forOp.getBody()->getTerminator()); resolveConflicts() local
714 auto yieldOp = cast<scf::YieldOp>(forOp.getBody()->getTerminator()); getBufferType() local
792 auto yieldOp = cast<scf::YieldOp>(forOp.getBody()->getTerminator()); verifyAnalysis() local
865 auto yieldOp = whileOp.getYieldOp(); bufferRelation() local
1030 auto yieldOp = whileOp.getYieldOp(); getBufferType() local
1089 auto yieldOp = whileOp.getYieldOp(); verifyAnalysis() local
1143 auto yieldOp = cast<scf::YieldOp>(op); bufferize() local
[all...]
H A DForToWhile.cpp89 for (auto yieldOp : afterBlock->getOps<scf::YieldOp>()) { in matchAndRewrite() local
H A DWrapInZeroTripCheck.cpp105 auto yieldOp = cast<scf::YieldOp>(newBlock->getTerminator()); in wrapWhileLoopInZeroTripCheck() local
/llvm-project/mlir/lib/Dialect/Vector/Transforms/
H A DBufferizableOpInterfaceImpl.cpp179 auto yieldOp = in getAliasingOpOperands() local
215 auto yieldOp = in bufferize() local
283 auto yieldOp = cast<vector::YieldOp>(op); in bufferize() local
/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparseIterationToScf.cpp111 auto yieldOp = matchAndRewrite() local
153 auto yieldOp = llvm::cast<sparse_tensor::YieldOp>( matchAndRewrite() local
H A DSparseTensorRewriting.cpp74 auto yieldOp = cast<linalg::YieldOp>(op.getRegion().front().getTerminator()); in isSampling() local
101 auto yieldOp = cast<linalg::YieldOp>(op.getRegion().front().getTerminator()); in isSumOfMul() local
114 auto yieldOp = cast<linalg::YieldOp>(op.getRegion().front().getTerminator()); in isZeroYield() local
H A DSparseGPUCodegen.cpp359 auto yieldOp = cast<linalg::YieldOp>(op.getRegion().front().getTerminator()); in matchSumOfMultOfArgs() local
374 auto yieldOp = cast<linalg::YieldOp>(op.getRegion().front().getTerminator()); in matchSumReductionOfMulUnary() local
/llvm-project/mlir/lib/Dialect/EmitC/Transforms/
H A DTransforms.cpp38 auto yieldOp = builder.create<emitc::YieldOp>(loc, result); in createExpression() local
/llvm-project/mlir/lib/Dialect/Tosa/Transforms/
H A DTosaInferShapes.cpp181 if (auto yieldOp = dyn_cast<YieldOp>(block.getTerminator())) propagateShapesToTosaWhile() local
191 for (auto yieldOp : yieldOps) { propagateShapesToTosaWhile() local
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DDecomposeLinalgOps.cpp167 if (auto yieldOp = dyn_cast<YieldOp>(user)) { in createPeeledGenericOp() local
305 auto *yieldOp = residualGenericOpBody->getTerminator(); in matchAndRewrite() local
H A DEraseUnusedOperandsAndResults.cpp44 auto yieldOp = dyn_cast<linalg::YieldOp>(argUserOp); in isResultValueDead() local
200 auto yieldOp = cast<YieldOp>(genericOp.getBody()->getTerminator()); in deduplicateOutputOperands() local
H A DConvertToDestinationStyle.cpp155 auto yieldOp = cast<tensor::YieldOp>(body->getTerminator()); in movePaddingToFillOrGenericOp() local
268 assert(isa<vector::YieldOp>(yieldOp) && "expected yield op terminator"); in bufferizeToAllocation() local
414 rewriter.replaceOpWithNewOp<linalg::YieldOp>(yieldOp, yieldOp in rewriteInDestinationPassingStyle() local
[all...]
H A DHoisting.cpp70 auto yieldOp = cast<scf::YieldOp>(loop.getBody()->getTerminator()); in replaceWithDifferentYield() local
H A DConstantFold.cpp281 auto yieldOp = dyn_cast<linalg::YieldOp>(body.getTerminator()); in getRegionComputeFn() local
H A DDetensorize.cpp74 YieldOp yieldOp = dyn_cast<YieldOp>(op.getRegion().back().getTerminator()); in matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/GPU/Transforms/
H A DAsyncRegionRewriter.cpp152 Operation *yieldOp = executeOp.getBody()->getTerminator(); in addExecuteResults() local
326 auto yieldOp = cast<async::YieldOp>(executeOp.getBody()->getTerminator()); in operator ()() local
/llvm-project/mlir/lib/Conversion/SCFToControlFlow/
H A DSCFToControlFlow.cpp574 auto yieldOp = cast<scf::YieldOp>(after->getTerminator()); matchAndRewrite() local
/llvm-project/mlir/lib/Conversion/SCFToOpenMP/
H A DSCFToOpenMP.cpp427 if (auto yieldOp = dyn_cast<omp::YieldOp>(*cloneOp)) { in matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/Linalg/IR/
H A DLinalgInterfaces.cpp95 auto yieldOp = dyn_cast<linalg::YieldOp>(body->back()); isaFillOpInterface() local
135 auto yieldOp = dyn_cast<linalg::YieldOp>(body->back()); isaElemwiseSingleUnaryOrBinaryOpInterface() local
/llvm-project/mlir/lib/Dialect/Affine/Transforms/
H A DSuperVectorize.cpp1453 static Operation *vectorizeAffineYieldOp(AffineYieldOp yieldOp, in vectorizeAffineYieldOp() argument
1515 if (auto yieldOp = dyn_cast<AffineYieldOp>(op)) in vectorizeOneOperation() local
/llvm-project/mlir/lib/Dialect/SCF/IR/
H A DSCF.cpp245 if (YieldOp yieldOp = dyn_cast<YieldOp>(blk.getTerminator())) { in matchAndRewrite() local
407 auto yieldOp = cast<scf::YieldOp>(getBody()->getTerminator()); in promoteIfSingleIteration() local
559 auto yieldOp = cast<scf::YieldOp>(getBody()->getTerminator()); in replaceWithAdditionalYields() local
1984 auto yieldOp = llvm::dyn_cast<YieldOp>(b.back()); inferReturnTypes() local
2220 auto yieldOp = cast<scf::YieldOp>(dest->getTerminator()); transferBody() local
3552 Operation *yieldOp = afterBlock.getTerminator(); matchAndRewrite() local
[all...]
/llvm-project/flang/lib/Optimizer/HLFIR/IR/
H A DHLFIROps.cpp1671 printYieldOpCleanup(mlir::OpAsmPrinter & p,YieldOp yieldOp,mlir::Region & cleanup) printYieldOpCleanup() argument
1696 hlfir::YieldOp yieldOp = verify() local
1713 hlfir::YieldOp yieldOp = getYieldOp() local
/llvm-project/flang/lib/Optimizer/HLFIR/Transforms/
H A DLowerHLFIROrderedAssignments.cpp1035 auto yieldOp = mlir::cast<hlfir::YieldOp>(terminator); generateMaskedExprCleanUps() local
1233 auto yieldOp = mlir::dyn_cast<hlfir::YieldOp>( rhsIsArray() local
/llvm-project/mlir/lib/Conversion/VectorToGPU/
H A DVectorToGPU.cpp1262 } else if (auto yieldOp = dyn_cast<scf::YieldOp>(op)) { convertVectorToMMAOps() local
1298 __anon1277acc40d02(scf::YieldOp yieldOp) convertVectorToNVVMCompatibleMMASync() argument

12