Lines Matching refs:conditionalBlock
53 auto *conditionalBlock = &loop.getRegion().front(); in matchAndRewrite() local
54 conditionalBlock->addArgument(rewriter.getIndexType(), loc); in matchAndRewrite()
56 rewriter.splitBlock(conditionalBlock, conditionalBlock->begin()); in matchAndRewrite()
89 rewriter.create<mlir::cf::BranchOp>(loc, conditionalBlock, loopOperands); in matchAndRewrite()
94 auto iv = conditionalBlock->getArgument(0); in matchAndRewrite()
98 auto lastArg = conditionalBlock->getNumArguments() - 1; in matchAndRewrite()
99 auto itersLeft = conditionalBlock->getArgument(lastArg); in matchAndRewrite()
110 rewriter.create<mlir::cf::BranchOp>(loc, conditionalBlock, loopCarried); in matchAndRewrite()
114 rewriter.setInsertionPointToEnd(conditionalBlock); in matchAndRewrite()
126 ? conditionalBlock->getArguments() in matchAndRewrite()
127 : conditionalBlock->getArguments().drop_front(); in matchAndRewrite()