Lines Matching refs:newForOp
1283 AffineForOp newForOp = mlir::replaceForOpWithNewYields( in loopUnrollJamByFactor() local
1285 newLoopsWithIterArgs.push_back(newForOp); in loopUnrollJamByFactor()
1288 forOp = newForOp; in loopUnrollJamByFactor()
1292 ValueRange newIterArgs = newForOp.getRegionIterArgs(); in loopUnrollJamByFactor()
1294 ValueRange newResults = newForOp.getResults(); in loopUnrollJamByFactor()
1338 for (auto newForOp : newLoopsWithIterArgs) { in loopUnrollJamByFactor() local
1340 newForOp.getNumIterOperands() / unrollJamFactor; in loopUnrollJamByFactor()
1341 unsigned numControlOperands = newForOp.getNumControlOperands(); in loopUnrollJamByFactor()
1342 auto yieldOp = cast<AffineYieldOp>(newForOp.getBody()->getTerminator()); in loopUnrollJamByFactor()
1350 newForOp.setOperand(numControlOperands + i * oldNumIterOperands + j, in loopUnrollJamByFactor()
1352 newForOp.getOperand(numControlOperands + j))); in loopUnrollJamByFactor()
1661 auto newForOp = b.create<AffineForOp>(t.getLoc(), lbOperands, lbMap, in stripmineSink() local
1666 newForOp.getBody()->getOperations().splice( in stripmineSink()
1667 newForOp.getBody()->getOperations().begin(), in stripmineSink()
1669 replaceAllUsesInRegionWith(iv, newForOp.getInductionVar(), in stripmineSink()
1670 newForOp.getRegion()); in stripmineSink()
1671 innerLoops.push_back(newForOp); in stripmineSink()