Lines Matching refs:ubOperands
255 auto ubOperands = srcForOp.getUpperBoundOperands(); in generateShiftedLoop() local
258 assert(ubMap.getNumInputs() == ubOperands.size()); in generateShiftedLoop()
261 ubOperands, ubMap, srcForOp.getStep()); in generateShiftedLoop()
619 SmallVector<Value, 4> lbOperands, ubOperands; in setIntraTileBoundsParametric() local
623 ubOperands.reserve(ub.getNumOperands() + 2); in setIntraTileBoundsParametric()
629 ubOperands.push_back(ub.getOperand(j)); in setIntraTileBoundsParametric()
634 ubOperands.push_back(newInterTileLoop.getInductionVar()); in setIntraTileBoundsParametric()
639 AffineExpr ubLoopIvExpr = b.getAffineDimExpr(ubOperands.size() - 1); in setIntraTileBoundsParametric()
645 ubOperands.push_back(ub.getOperand(origUbMap.getNumDims() + j)); in setIntraTileBoundsParametric()
649 ubOperands.push_back(tileSize); in setIntraTileBoundsParametric()
685 newIntraTileLoop.setUpperBound(ubOperands, ubMap); in setIntraTileBoundsParametric()
719 SmallVector<Value, 4> ubOperands; in setInterTileBoundsParametric() local
721 ubOperands.reserve(ub.getNumOperands() + 1); in setInterTileBoundsParametric()
724 ubOperands.push_back(ub.getOperand(j)); in setInterTileBoundsParametric()
728 ubOperands.push_back(ub.getOperand(origUbMap.getNumDims() + j)); in setInterTileBoundsParametric()
731 ubOperands.push_back(tileSize); in setInterTileBoundsParametric()
782 newLoop.setUpperBound(ubOperands, ubMap); in setInterTileBoundsParametric()
863 SmallVector<Value, 4> ubOperands; in constructTiledIndexSetHyperRect() local
865 ubOperands.reserve(ub.getNumOperands() + 1); in constructTiledIndexSetHyperRect()
868 ubOperands.push_back(ub.getOperand(j)); in constructTiledIndexSetHyperRect()
871 ubOperands.push_back(newLoops[i].getInductionVar()); in constructTiledIndexSetHyperRect()
875 ubOperands.push_back(ub.getOperand(origUbMap.getNumDims() + j)); in constructTiledIndexSetHyperRect()
888 newLoops[width + i].setUpperBound(/*operands=*/ubOperands, ubMap); in constructTiledIndexSetHyperRect()
1664 SmallVector<Value, 4> ubOperands(forOp.getUpperBoundOperands()); in stripmineSink() local
1665 augmentMapAndBounds(b, forOp.getInductionVar(), &ubMap, &ubOperands, in stripmineSink()
1674 ubOperands, ubMap, originalStep); in stripmineSink()
2221 ArrayRef<AffineMap> ubMaps, ArrayRef<Value> ubOperands, in generatePointWiseCopy() argument
2228 return ubMap.getNumInputs() == ubOperands.size(); in generatePointWiseCopy()
2242 ubOperands, ubMaps[d]); in generatePointWiseCopy()
2909 ValueRange ubOperands, AffineMap ubMap, int64_t step) { in createCanonicalizedAffineForOp() argument
2911 SmallVector<Value, 4> upperOperands(ubOperands); in createCanonicalizedAffineForOp()