Lines Matching refs:newBlockTransferArgs
623 SmallVector<Value, 4> newBlockTransferArgs, newIterArgs, newYieldValues, in matchAndRewrite() local
625 newBlockTransferArgs.reserve(1 + forOp.getNumIterOperands()); in matchAndRewrite()
626 newBlockTransferArgs.push_back(Value()); // iv placeholder with null value in matchAndRewrite()
648 newBlockTransferArgs.push_back(std::get<0>(it)); in matchAndRewrite()
654 newBlockTransferArgs.push_back(Value()); // placeholder with null value in matchAndRewrite()
668 newBlockTransferArgs[0] = newBlock.getArgument(0); // iv in matchAndRewrite()
671 Value &blockTransferArg = newBlockTransferArgs[1 + idx]; in matchAndRewrite()
682 assert(oldBlock.getNumArguments() == newBlockTransferArgs.size() && in matchAndRewrite()
690 rewriter.mergeBlockBefore(&oldBlock, newYieldOp, newBlockTransferArgs); in matchAndRewrite()
709 rewriter.mergeBlocks(&oldBlock, &newBlock, newBlockTransferArgs); in matchAndRewrite()
809 SmallVector<Value, 4> newBlockTransferArgs(newBlock.getArguments().begin(), in replaceTensorCastForOpIterArg() local
820 newBlockTransferArgs[newRegionIterArg.getArgNumber()] = castIn; in replaceTensorCastForOpIterArg()
824 rewriter.mergeBlocks(&oldBlock, &newBlock, newBlockTransferArgs); in replaceTensorCastForOpIterArg()