Lines Matching refs:lbOperands

171       auto lbOperands = forOp.getLowerBoundOperands();  in promoteIfSingleIteration()  local
176 iv.replaceAllUsesWith(lbOperands[0]); in promoteIfSingleIteration()
179 builder.create<AffineApplyOp>(forOp.getLoc(), lbMap, lbOperands); in promoteIfSingleIteration()
254 auto lbOperands = srcForOp.getLowerBoundOperands(); in generateShiftedLoop() local
257 assert(lbMap.getNumInputs() == lbOperands.size()); in generateShiftedLoop()
260 auto loopChunk = b.create<AffineForOp>(srcForOp.getLoc(), lbOperands, lbMap, in generateShiftedLoop()
619 SmallVector<Value, 4> lbOperands, ubOperands; in setIntraTileBoundsParametric() local
622 lbOperands.reserve(lb.getNumOperands() + 2); in setIntraTileBoundsParametric()
627 lbOperands.push_back(lb.getOperand(j)); in setIntraTileBoundsParametric()
633 lbOperands.push_back(newInterTileLoop.getInductionVar()); in setIntraTileBoundsParametric()
638 AffineExpr lbLoopIvExpr = b.getAffineDimExpr(lbOperands.size() - 1); in setIntraTileBoundsParametric()
643 lbOperands.push_back(lb.getOperand(origLbMap.getNumDims() + j)); in setIntraTileBoundsParametric()
648 lbOperands.push_back(tileSize); in setIntraTileBoundsParametric()
680 newIntraTileLoop.setLowerBound(lbOperands, lbMap); in setIntraTileBoundsParametric()
1659 SmallVector<Value, 4> lbOperands(forOp.getLowerBoundOperands()); in stripmineSink() local
1660 augmentMapAndBounds(b, forOp.getInductionVar(), &lbMap, &lbOperands); in stripmineSink()
1673 auto newForOp = b.create<AffineForOp>(t.getLoc(), lbOperands, lbMap, in stripmineSink()
2220 ArrayRef<AffineMap> lbMaps, ArrayRef<Value> lbOperands, in generatePointWiseCopy() argument
2225 return lbMap.getNumInputs() == lbOperands.size(); in generatePointWiseCopy()
2241 auto forOp = createCanonicalizedAffineForOp(b, loc, lbOperands, lbMaps[d], in generatePointWiseCopy()
2249 AffineMap::get(lbOperands.size(), 0, fastBufOffsets[d]); in generatePointWiseCopy()
2250 auto offset = b.create<AffineApplyOp>(loc, fastBufOffsetMap, lbOperands); in generatePointWiseCopy()
2908 OpBuilder b, Location loc, ValueRange lbOperands, AffineMap lbMap, in createCanonicalizedAffineForOp() argument
2910 SmallVector<Value, 4> lowerOperands(lbOperands); in createCanonicalizedAffineForOp()