Home
last modified time | relevance | path

Searched refs:yieldOperands (Results 1 – 7 of 7) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/SCF/Transforms/
H A DForToWhile.cpp84 SmallVector<Value> yieldOperands = yieldOp.getOperands(); in matchAndRewrite() local
85 yieldOperands.insert(yieldOperands.begin(), ivIncOp.getResult()); in matchAndRewrite()
86 yieldOp->setOperands(yieldOperands); in matchAndRewrite()
H A DLoopPipelining.cpp341 llvm::SmallVector<Value> yieldOperands; in createKernel() local
343 yieldOperands.push_back(mapping.lookupOrDefault(retVal)); in createKernel()
352 setValueMapping(it.first, newForOp->getResult(yieldOperands.size()), in createKernel()
354 yieldOperands.push_back( in createKernel()
355 newForOp.getBody()->getArguments()[yieldOperands.size() + 1 + in createKernel()
358 setValueMapping(it.first, newForOp->getResult(yieldOperands.size()), in createKernel()
360 yieldOperands.push_back(mapping.lookupOrDefault(it.first)); in createKernel()
372 rewriter.create<scf::YieldOp>(forOp.getLoc(), yieldOperands); in createKernel()
/llvm-project-15.0.7/mlir/lib/Conversion/SCFToControlFlow/
H A DSCFToControlFlow.cpp487 SmallVector<Value, 4> yieldOperands; in matchAndRewrite() local
488 yieldOperands.reserve(parallelOp.getNumResults()); in matchAndRewrite()
495 Value arg = iterArgs[yieldOperands.size()]; in matchAndRewrite()
496 yieldOperands.push_back(reduceBlock.getTerminator()->getOperand(0)); in matchAndRewrite()
513 if (!yieldOperands.empty()) { in matchAndRewrite()
515 rewriter.create<scf::YieldOp>(loc, yieldOperands); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Conversion/VectorToGPU/
H A DVectorToGPU.cpp782 auto yieldOperands = llvm::to_vector<4>(op.getOperands()); in convertYieldOp() local
789 yieldOperands[operand.index()] = loop.getIterOperands()[operand.index()]; in convertYieldOp()
790 yieldOperands.push_back(it->second); in convertYieldOp()
792 b.create<scf::YieldOp>(op.getLoc(), yieldOperands); in convertYieldOp()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorDistribute.cpp821 SmallVector<Value> yieldOperands; in matchAndRewrite() local
823 yieldOperands.push_back(operand); in matchAndRewrite()
827 rewriter.create<vector::YieldOp>(innerWarp.getLoc(), yieldOperands); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/SCF/IR/
H A DSCF.cpp769 auto yieldOperands = yieldOp.getOperands(); in matchAndRewrite() local
770 if (llvm::any_of(yieldOperands, in matchAndRewrite()
773 rewriter.replaceOp(op, yieldOperands); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/IR/
H A DAffineOps.cpp2264 auto yieldOperands = llvm::to_vector<4>(yield.getOperands()); in replaceForOpWithNewYields() local
2265 yieldOperands.append(newYieldedValues.begin(), newYieldedValues.end()); in replaceForOpWithNewYields()
2266 b.create<AffineYieldOp>(yield.getLoc(), yieldOperands); in replaceForOpWithNewYields()