Lines Matching refs:rootLoop
737 void finishVectorizationPattern(AffineForOp rootLoop);
897 void VectorizationState::finishVectorizationPattern(AffineForOp rootLoop) { in finishVectorizationPattern() argument
899 eraseLoopNest(rootLoop); in finishVectorizationPattern()
1533 AffineForOp rootLoop = loops[0][0]; in vectorizeLoopNest() local
1534 VectorizationState state(rootLoop.getContext()); in vectorizeLoopNest()
1535 state.builder.setInsertionPointAfter(rootLoop); in vectorizeLoopNest()
1545 if (!isVectorizableLoopBody(rootLoop, vectorTransferPattern())) { in vectorizeLoopNest()
1558 auto opVecResult = rootLoop.walk<WalkOrder::PreOrder>([&](Operation *op) { in vectorizeLoopNest()
1573 << rootLoop << "\n"); in vectorizeLoopNest()
1575 auto vecRootLoopIt = state.opVectorReplacement.find(rootLoop); in vectorizeLoopNest()
1587 assert(state.opVectorReplacement.count(rootLoop) == 1 && in vectorizeLoopNest()
1591 << *state.opVectorReplacement[rootLoop]); in vectorizeLoopNest()
1594 state.finishVectorizationPattern(rootLoop); in vectorizeLoopNest()