Lines Matching defs:newForOp

1182     AffineForOp newForOp =
1188     newLoopsWithIterArgs.push_back(newForOp);
1191       forOp = newForOp;
1192     // Update `operandMaps` for `newForOp` iterArgs and results.
1193     ValueRange newIterArgs = newForOp.getRegionIterArgs();
1195     ValueRange newResults = newForOp.getResults();
1201         // `newForOp` has `unrollJamFactor` - 1 new sets of iterArgs and
1239     for (auto newForOp : newLoopsWithIterArgs) {
1241           newForOp.getNumIterOperands() / unrollJamFactor;
1242       unsigned numControlOperands = newForOp.getNumControlOperands();
1243       auto yieldOp = cast<AffineYieldOp>(newForOp.getBody()->getTerminator());
1251         newForOp.setOperand(numControlOperands + i * oldNumIterOperands + j,
1253                                 newForOp.getOperand(numControlOperands + j)));
1556     // Insert newForOp before the terminator of `t`.
1558     auto newForOp = b.create<AffineForOp>(t.getLoc(), lbOperands, lbMap,
1561     // Skip terminator and `newForOp` which is just before the terminator.
1563     newForOp.getBody()->getOperations().splice(
1564         newForOp.getBody()->getOperations().begin(),
1566     replaceAllUsesInRegionWith(iv, newForOp.getInductionVar(),
1567                                newForOp.getRegion());
1568     innerLoops.push_back(newForOp);