Lines Matching refs:getBody
123 auto innerResults = forOp.getBody()->getTerminator()->getOperands(); in replaceIterArgsAndYieldResults()
144 OpBuilder topBuilder(forOp->getParentOfType<func::FuncOp>().getBody()); in promoteIfSingleIteration()
167 forOp.getBody()->back().erase(); in promoteIfSingleIteration()
169 forOp.getBody()->getOperations()); in promoteIfSingleIteration()
198 auto bodyBuilder = OpBuilder::atBlockTerminator(loopChunk.getBody()); in generateShiftedLoop()
238 assert(forOp.getBody()->getOperations().size() == shifts.size() && in affineForOpBodySkew()
240 if (forOp.getBody()->begin() == std::prev(forOp.getBody()->end())) in affineForOpBodySkew()
274 for (auto &op : forOp.getBody()->without_terminator()) { in affineForOpBodySkew()
495 auto &ops = src.getBody()->getOperations(); in moveLoopBodyImpl()
496 dest.getBody()->getOperations().splice(loc, ops, ops.begin(), in moveLoopBodyImpl()
503 moveLoopBodyImpl(src, dest, dest.getBody()->begin()); in moveLoopBody()
522 pointLoop.getBody()->getOperations().splice( in constructTiledLoopNest()
523 pointLoop.getBody()->begin(), topLoop->getBlock()->getOperations(), in constructTiledLoopNest()
536 tileSpaceLoop.getBody()->getOperations().splice( in constructTiledLoopNest()
537 tileSpaceLoop.getBody()->begin(), topLoop->getBlock()->getOperations(), in constructTiledLoopNest()
1105 if (llvm::hasSingleElement(forOp.getBody()->getOperations())) in loopUnrollByFactor()
1128 auto yieldedValues = forOp.getBody()->getTerminator()->getOperands(); in loopUnrollByFactor()
1134 forOp.getBody(), forOp.getInductionVar(), unrollFactor, in loopUnrollByFactor()
1213 if (llvm::hasSingleElement(forOp.getBody()->getOperations())) in loopUnrollJamByFactor()
1273 cast<AffineYieldOp>(oldForOp.getBody()->getTerminator()).getOperands(); in loopUnrollJamByFactor()
1342 auto yieldOp = cast<AffineYieldOp>(newForOp.getBody()->getTerminator()); in loopUnrollJamByFactor()
1395 assert(&*forOpA.getBody()->begin() == forOpB.getOperation()); in interchangeLoops()
1396 auto &forOpABody = forOpA.getBody()->getOperations(); in interchangeLoops()
1397 auto &forOpBBody = forOpB.getBody()->getOperations(); in interchangeLoops()
1478 if (parentForOp != enclosingLoop || !hasTwoElements(parentForOp.getBody())) in isPerfectlyNested()
1514 auto *destBody = input[invPermMap.back().second].getBody(); in permuteLoops()
1515 auto *srcBody = input.back().getBody(); in permuteLoops()
1545 auto *destBody = input[parentPosInInput].getBody(); in permuteLoops()
1660 auto b = OpBuilder::atBlockTerminator(t.getBody()); in stripmineSink()
1663 auto begin = t.getBody()->begin(); in stripmineSink()
1665 auto nOps = t.getBody()->getOperations().size() - 2; in stripmineSink()
1666 newForOp.getBody()->getOperations().splice( in stripmineSink()
1667 newForOp.getBody()->getOperations().begin(), in stripmineSink()
1668 t.getBody()->getOperations(), begin, std::next(begin, nOps)); in stripmineSink()
1774 builder.setInsertionPointToStart(outermost.getBody()); in coalesceLoops()
1821 innermost.getBody()->back().erase(); in coalesceLoops()
1822 outermost.getBody()->getOperations().splice( in coalesceLoops()
1824 innermost.getBody()->getOperations()); in coalesceLoops()
1975 b = OpBuilder::atBlockTerminator(forOp.getBody()); in generatePointWiseCopy()
2048 OpBuilder topBuilder(f.getBody()); in generateCopy()
2066 OpBuilder top(func.getBody()); in generateCopy()
2562 return affineDataCopyGenerate(forOp.getBody()->begin(), in affineDataCopyGenerate()
2563 std::prev(forOp.getBody()->end()), copyOptions, in affineDataCopyGenerate()
2605 gatherLoopsInBlock(forOp.getBody(), currLoopDepth + 1, depthToLoops); in gatherLoopsInBlock()
2768 b = OpBuilder::atBlockTerminator(fullTileLoop.getBody()); in createFullTiles()
2777 b = OpBuilder::atBlockTerminator(fullTileLoops.back().getBody()); in createFullTiles()
2778 for (auto &op : inputNest.back().getBody()->without_terminator()) in createFullTiles()