Lines Matching refs:lbOperands
149 auto lbOperands = forOp.getLowerBoundOperands(); in promoteIfSingleIteration() local
154 iv.replaceAllUsesWith(lbOperands[0]); in promoteIfSingleIteration()
157 builder.create<AffineApplyOp>(forOp.getLoc(), lbMap, lbOperands); in promoteIfSingleIteration()
185 auto lbOperands = srcForOp.getLowerBoundOperands(); in generateShiftedLoop() local
188 assert(lbMap.getNumInputs() == lbOperands.size()); in generateShiftedLoop()
191 auto loopChunk = b.create<AffineForOp>(srcForOp.getLoc(), lbOperands, lbMap, in generateShiftedLoop()
570 SmallVector<Value, 4> lbOperands, ubOperands; in setIntraTileBoundsParametric() local
573 lbOperands.reserve(lb.getNumOperands() + 2); in setIntraTileBoundsParametric()
578 lbOperands.push_back(lb.getOperand(j)); in setIntraTileBoundsParametric()
584 lbOperands.push_back(newInterTileLoop.getInductionVar()); in setIntraTileBoundsParametric()
589 AffineExpr lbLoopIvExpr = b.getAffineDimExpr(lbOperands.size() - 1); in setIntraTileBoundsParametric()
594 lbOperands.push_back(lb.getOperand(origLbMap.getNumDims() + j)); in setIntraTileBoundsParametric()
599 lbOperands.push_back(tileSize); in setIntraTileBoundsParametric()
631 newIntraTileLoop.setLowerBound(lbOperands, lbMap); in setIntraTileBoundsParametric()
1647 SmallVector<Value, 4> lbOperands(forOp.getLowerBoundOperands()); in stripmineSink() local
1648 augmentMapAndBounds(b, forOp.getInductionVar(), &lbMap, &lbOperands); in stripmineSink()
1661 auto newForOp = b.create<AffineForOp>(t.getLoc(), lbOperands, lbMap, in stripmineSink()
1949 ArrayRef<AffineMap> lbMaps, ArrayRef<Value> lbOperands, in generatePointWiseCopy() argument
1954 return lbMap.getNumInputs() == lbOperands.size(); in generatePointWiseCopy()
1970 auto forOp = createCanonicalizedAffineForOp(b, loc, lbOperands, lbMaps[d], in generatePointWiseCopy()
1978 AffineMap::get(lbOperands.size(), 0, fastBufOffsets[d]); in generatePointWiseCopy()
1979 auto offset = b.create<AffineApplyOp>(loc, fastBufOffsetMap, lbOperands); in generatePointWiseCopy()
2628 OpBuilder b, Location loc, ValueRange lbOperands, AffineMap lbMap, in createCanonicalizedAffineForOp() argument
2630 SmallVector<Value, 4> lowerOperands(lbOperands); in createCanonicalizedAffineForOp()