Lines Matching refs:newBlock
665 Block &newBlock = newForOp.getRegion().front(); in matchAndRewrite() local
668 newBlockTransferArgs[0] = newBlock.getArgument(0); // iv in matchAndRewrite()
689 auto newYieldOp = cast<scf::YieldOp>(newBlock.getTerminator()); in matchAndRewrite()
691 rewriter.eraseOp(newBlock.getTerminator()->getPrevNode()); in matchAndRewrite()
709 rewriter.mergeBlocks(&oldBlock, &newBlock, newBlockTransferArgs); in matchAndRewrite()
710 auto mergedYieldOp = cast<scf::YieldOp>(newBlock.getTerminator()); in matchAndRewrite()
808 Block &newBlock = newForOp.getRegion().front(); in replaceTensorCastForOpIterArg() local
809 SmallVector<Value, 4> newBlockTransferArgs(newBlock.getArguments().begin(), in replaceTensorCastForOpIterArg()
810 newBlock.getArguments().end()); in replaceTensorCastForOpIterArg()
815 rewriter.setInsertionPoint(&newBlock, newBlock.begin()); in replaceTensorCastForOpIterArg()
824 rewriter.mergeBlocks(&oldBlock, &newBlock, newBlockTransferArgs); in replaceTensorCastForOpIterArg()
827 auto clonedYieldOp = cast<scf::YieldOp>(newBlock.getTerminator()); in replaceTensorCastForOpIterArg()