Lines Matching defs:ubOperands

179   auto ubOperands = srcForOp.getUpperBoundOperands();
182 assert(ubMap.getNumInputs() == ubOperands.size());
185 b.create<AffineForOp>(srcForOp.getLoc(), lbOperands, lbMap, ubOperands,
485 SmallVector<Value, 4> lbOperands, ubOperands;
489 ubOperands.reserve(ub.getNumOperands() + 2);
495 ubOperands.push_back(ub.getOperand(j));
497 // Add a new dim operand in lb/ubOperands corresponding to the origLoop
500 ubOperands.push_back(newInterTileLoop.getInductionVar());
503 // lb/ubOperands is guaranteed to be atleast one.
505 AffineExpr ubLoopIvExpr = b.getAffineDimExpr(ubOperands.size() - 1);
511 ubOperands.push_back(ub.getOperand(origUbMap.getNumDims() + j));
515 ubOperands.push_back(tileSize);
551 newIntraTileLoop.setUpperBound(ubOperands, ubMap);
585 SmallVector<Value, 4> ubOperands;
587 ubOperands.reserve(ub.getNumOperands() + 1);
590 ubOperands.push_back(ub.getOperand(j));
594 ubOperands.push_back(ub.getOperand(origUbMap.getNumDims() + j));
597 ubOperands.push_back(tileSize);
648 newLoop.setUpperBound(ubOperands, ubMap);
735 SmallVector<Value, 4> ubOperands;
737 ubOperands.reserve(ub.getNumOperands() + 1);
740 ubOperands.push_back(ub.getOperand(j));
743 ubOperands.push_back(newLoops[i].getInductionVar());
747 ubOperands.push_back(ub.getOperand(origUbMap.getNumDims() + j));
760 newLoops[width + i].setUpperBound(/*operands=*/ubOperands, ubMap);
1549 SmallVector<Value, 4> ubOperands(forOp.getUpperBoundOperands());
1550 augmentMapAndBounds(b, forOp.getInductionVar(), &ubMap, &ubOperands,
1559 ubOperands, ubMap, originalStep);
1629 SmallVector<Value, 4> ubOperands(ub.getOperands().begin(),
1635 prev = builder.create<AffineMinOp>(loc, origUbMap, ubOperands);
1637 prev = builder.create<AffineApplyOp>(loc, origUbMap, ubOperands);
1645 ubOperands = ub.getOperands();
1649 upperBound = builder.create<AffineMinOp>(loc, origUbMap, ubOperands);
1651 upperBound = builder.create<AffineApplyOp>(loc, origUbMap, ubOperands);
1833 /// `ubMaps` along with `lbOperands` and `ubOperands` hold the lower and upper
1848 ArrayRef<AffineMap> ubMaps, ArrayRef<Value> ubOperands,
1855 return ubMap.getNumInputs() == ubOperands.size();
1869 ubOperands, ubMaps[d]);
2128 /*ubOperands=*/regionSymbols, fastBufOffsets,
2531 ValueRange ubOperands, AffineMap ubMap, int64_t step) {
2533 SmallVector<Value, 4> upperOperands(ubOperands);