Lines Matching defs:rootLoop
742 void finishVectorizationPattern(AffineForOp rootLoop);
895 void VectorizationState::finishVectorizationPattern(AffineForOp rootLoop) {
897 eraseLoopNest(rootLoop);
1565 AffineForOp rootLoop = loops[0][0];
1566 VectorizationState state(rootLoop.getContext());
1567 state.builder.setInsertionPointAfter(rootLoop);
1577 if (!isVectorizableLoopBody(rootLoop, vectorTransferPattern())) {
1590 auto opVecResult = rootLoop.walk<WalkOrder::PreOrder>([&](Operation *op) {
1605 << rootLoop << "\n");
1607 auto vecRootLoopIt = state.opVectorReplacement.find(rootLoop);
1619 assert(state.opVectorReplacement.count(rootLoop) == 1 &&
1623 << *state.opVectorReplacement[rootLoop]);
1626 state.finishVectorizationPattern(rootLoop);