Lines Matching refs:newForOp

661     scf::ForOp newForOp = rewriter.create<scf::ForOp>(  in matchAndRewrite()  local
664 newForOp->setAttrs(forOp->getAttrs()); in matchAndRewrite()
665 Block &newBlock = newForOp.getRegion().front(); in matchAndRewrite()
676 blockTransferArg = newForOp.getRegionIterArgs()[collapsedIdx]; in matchAndRewrite()
677 newResultVal = newForOp.getResult(collapsedIdx++); in matchAndRewrite()
804 scf::ForOp newForOp = rewriter.create<scf::ForOp>( in replaceTensorCastForOpIterArg() local
807 newForOp->setAttrs(forOp->getAttrs()); in replaceTensorCastForOpIterArg()
808 Block &newBlock = newForOp.getRegion().front(); in replaceTensorCastForOpIterArg()
816 BlockArgument newRegionIterArg = newForOp.getRegionIterArgForOpOperand( in replaceTensorCastForOpIterArg()
817 newForOp->getOpOperand(operand.getOperandNumber())); in replaceTensorCastForOpIterArg()
818 Value castIn = rewriter.create<tensor::CastOp>(newForOp.getLoc(), oldType, in replaceTensorCastForOpIterArg()
832 newForOp.getLoc(), newType, clonedYieldOp.getOperand(yieldIdx)); in replaceTensorCastForOpIterArg()
835 rewriter.create<scf::YieldOp>(newForOp.getLoc(), newYieldOperands); in replaceTensorCastForOpIterArg()
839 rewriter.setInsertionPointAfter(newForOp); in replaceTensorCastForOpIterArg()
840 SmallVector<Value> newResults = newForOp.getResults(); in replaceTensorCastForOpIterArg()
842 newForOp.getLoc(), oldType, newResults[yieldIdx]); in replaceTensorCastForOpIterArg()
844 return newForOp; in replaceTensorCastForOpIterArg()
896 auto newForOp = replaceTensorCastForOpIterArg(rewriter, iterOpOperand, in matchAndRewrite() local
900 rewriter.setInsertionPointAfter(newForOp); in matchAndRewrite()
901 SmallVector<Value> replacements = newForOp.getResults(); in matchAndRewrite()