Lines Matching refs:ubOperands

186   auto ubOperands = srcForOp.getUpperBoundOperands();  in generateShiftedLoop()  local
189 assert(ubMap.getNumInputs() == ubOperands.size()); in generateShiftedLoop()
192 ubOperands, ubMap, srcForOp.getStep()); in generateShiftedLoop()
570 SmallVector<Value, 4> lbOperands, ubOperands; in setIntraTileBoundsParametric() local
574 ubOperands.reserve(ub.getNumOperands() + 2); in setIntraTileBoundsParametric()
580 ubOperands.push_back(ub.getOperand(j)); in setIntraTileBoundsParametric()
585 ubOperands.push_back(newInterTileLoop.getInductionVar()); in setIntraTileBoundsParametric()
590 AffineExpr ubLoopIvExpr = b.getAffineDimExpr(ubOperands.size() - 1); in setIntraTileBoundsParametric()
596 ubOperands.push_back(ub.getOperand(origUbMap.getNumDims() + j)); in setIntraTileBoundsParametric()
600 ubOperands.push_back(tileSize); in setIntraTileBoundsParametric()
636 newIntraTileLoop.setUpperBound(ubOperands, ubMap); in setIntraTileBoundsParametric()
670 SmallVector<Value, 4> ubOperands; in setInterTileBoundsParametric() local
672 ubOperands.reserve(ub.getNumOperands() + 1); in setInterTileBoundsParametric()
675 ubOperands.push_back(ub.getOperand(j)); in setInterTileBoundsParametric()
679 ubOperands.push_back(ub.getOperand(origUbMap.getNumDims() + j)); in setInterTileBoundsParametric()
682 ubOperands.push_back(tileSize); in setInterTileBoundsParametric()
733 newLoop.setUpperBound(ubOperands, ubMap); in setInterTileBoundsParametric()
819 SmallVector<Value, 4> ubOperands; in constructTiledIndexSetHyperRect() local
821 ubOperands.reserve(ub.getNumOperands() + 1); in constructTiledIndexSetHyperRect()
824 ubOperands.push_back(ub.getOperand(j)); in constructTiledIndexSetHyperRect()
827 ubOperands.push_back(newLoops[i].getInductionVar()); in constructTiledIndexSetHyperRect()
831 ubOperands.push_back(ub.getOperand(origUbMap.getNumDims() + j)); in constructTiledIndexSetHyperRect()
844 newLoops[width + i].setUpperBound(/*operands=*/ubOperands, ubMap); in constructTiledIndexSetHyperRect()
1652 SmallVector<Value, 4> ubOperands(forOp.getUpperBoundOperands()); in stripmineSink() local
1653 augmentMapAndBounds(b, forOp.getInductionVar(), &ubMap, &ubOperands, in stripmineSink()
1662 ubOperands, ubMap, originalStep); in stripmineSink()
1732 SmallVector<Value, 4> ubOperands(ub.getOperands().begin(), in coalesceLoops() local
1738 prev = builder.create<AffineMinOp>(loc, origUbMap, ubOperands); in coalesceLoops()
1740 prev = builder.create<AffineApplyOp>(loc, origUbMap, ubOperands); in coalesceLoops()
1748 ubOperands = ub.getOperands(); in coalesceLoops()
1752 upperBound = builder.create<AffineMinOp>(loc, origUbMap, ubOperands); in coalesceLoops()
1754 upperBound = builder.create<AffineApplyOp>(loc, origUbMap, ubOperands); in coalesceLoops()
1950 ArrayRef<AffineMap> ubMaps, ArrayRef<Value> ubOperands, in generatePointWiseCopy() argument
1957 return ubMap.getNumInputs() == ubOperands.size(); in generatePointWiseCopy()
1971 ubOperands, ubMaps[d]); in generatePointWiseCopy()
2629 ValueRange ubOperands, AffineMap ubMap, int64_t step) { in createCanonicalizedAffineForOp() argument
2631 SmallVector<Value, 4> upperOperands(ubOperands); in createCanonicalizedAffineForOp()