Lines Matching refs:ubs

133                          SmallVectorImpl<Value> &ubs,  in unpackRanges()  argument
137 ubs.emplace_back(range.size); in unpackRanges()
526 SmallVector<Value, 4> lbs, ubs, steps; in doit() local
527 unpackRanges(loopRanges, lbs, ubs, steps); in doit()
529 b, loc, lbs, ubs, steps, iterArgInitValues, in doit()
569 SmallVector<Value, 4> lbs, ubs, steps; in doit() local
570 unpackRanges(loopRanges, lbs, ubs, steps); in doit()
581 mlir::buildAffineLoopNest(b, loc, lbs, ubs, constantSteps, in doit()
611 OpBuilder &b, Location loc, ValueRange lbs, ValueRange ubs, in generateParallelLoopNest() argument
616 assert(lbs.size() == ubs.size());
637 b, loc, lbs.take_front(), ubs.take_front(), steps.take_front(),
640 generateParallelLoopNest(b, loc, lbs.drop_front(), ubs.drop_front(), in __anon19d834b50d02()
651 loc, lbs.take_front(nOuterPar), ubs.take_front(nOuterPar),
657 ubs.drop_front(nOuterPar), steps.drop_front(nOuterPar), in __anon19d834b50e02()
680 loc, lbs.take_front(numProcessed), ubs.take_front(numProcessed),
686 ubs.drop_front(numProcessed), steps.drop_front(numProcessed), in __anon19d834b50f02()
697 Value cond = ab.slt(lbs[0], ubs[0]);
699 cond = ab._and(cond, ab.slt(lbs[i], ubs[i]));
703 b, loc, lbs.drop_front(numProcessed), ubs.drop_front(numProcessed), in __anon19d834b51002()
716 b, loc, lbs.drop_front(numProcessed), ubs.drop_front(numProcessed),
777 ValueRange lbs(lbsStorage), ubs(ubsStorage), steps(stepsStorage); in doit() local
779 b, loc, lbs, ubs, steps, iteratorTypes, in doit()
801 ValueRange ubs, ValueRange subShapeSizes, in makeTiledShape() argument
880 auto maxIndices = llvm::to_vector<8>(llvm::map_range(ubs, [&](Value ub) { in makeTiledShape()