Lines Matching refs:forOp
60 LogicalResult runOnAffineForOp(AffineForOp forOp);
76 f.walk([&](AffineForOp forOp) { in gatherInnermostLoops() argument
77 if (isInnermostAffineForOp(forOp)) in gatherInnermostLoops()
78 loops.push_back(forOp); in gatherInnermostLoops()
94 getOperation().walk([&](AffineForOp forOp) { in runOnOperation() argument
95 Optional<uint64_t> tripCount = getConstantTripCount(forOp); in runOnOperation()
97 loops.push_back(forOp); in runOnOperation()
99 for (auto forOp : loops) in runOnOperation() local
100 (void)loopUnrollFull(forOp); in runOnOperation()
112 for (auto forOp : loops) in runOnOperation() local
113 unrolled |= succeeded(runOnAffineForOp(forOp)); in runOnOperation()
122 LogicalResult LoopUnroll::runOnAffineForOp(AffineForOp forOp) { in runOnAffineForOp() argument
125 return loopUnrollByFactor(forOp, getUnrollFactor(forOp)); in runOnAffineForOp()
128 return loopUnrollFull(forOp); in runOnAffineForOp()
131 return loopUnrollUpToFactor(forOp, unrollFactor); in runOnAffineForOp()
132 return loopUnrollByFactor(forOp, unrollFactor); in runOnAffineForOp()