Searched refs:newBlockTransferArgs (Results 1 – 1 of 1) sorted by relevance
623 SmallVector<Value, 4> newBlockTransferArgs, newIterArgs, newYieldValues, in matchAndRewrite() local625 newBlockTransferArgs.reserve(1 + forOp.getNumIterOperands()); 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()820 newBlockTransferArgs[newRegionIterArg.getArgNumber()] = castIn; in replaceTensorCastForOpIterArg()[all …]