Lines Matching refs:AffineForOp
40 const std::function<unsigned(AffineForOp)> getUnrollFactor;
49 const std::function<unsigned(AffineForOp)> &getUnrollFactor = nullptr) in LoopUnroll()
60 LogicalResult runOnAffineForOp(AffineForOp forOp);
65 static bool isInnermostAffineForOp(AffineForOp op) { in isInnermostAffineForOp()
67 ->walk([&](AffineForOp nestedForOp) { in isInnermostAffineForOp()
75 SmallVectorImpl<AffineForOp> &loops) { in gatherInnermostLoops()
76 f.walk([&](AffineForOp forOp) { in gatherInnermostLoops()
89 SmallVector<AffineForOp, 4> loops; in runOnOperation()
94 getOperation().walk([&](AffineForOp forOp) { in runOnOperation()
105 SmallVector<AffineForOp, 4> loops; in runOnOperation()
122 LogicalResult LoopUnroll::runOnAffineForOp(AffineForOp forOp) { in runOnAffineForOp()
137 const std::function<unsigned(AffineForOp)> &getUnrollFactor) { in createLoopUnrollPass()