Lines Matching refs:lbs
132 static void unpackRanges(ArrayRef<Range> ranges, SmallVectorImpl<Value> &lbs, in unpackRanges() argument
136 lbs.emplace_back(range.offset); 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());
617 assert(lbs.size() == steps.size());
618 assert(lbs.size() == iteratorTypes.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),
656 nestedBuilder, nestedLoc, lbs.drop_front(nOuterPar), in __anon19d834b50e02()
680 loc, lbs.take_front(numProcessed), ubs.take_front(numProcessed),
685 nestedBuilder, nestedLoc, lbs.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]));
700 ivStorage.append(lbs.begin(), std::next(lbs.begin(), numProcessed));
703 b, loc, lbs.drop_front(numProcessed), ubs.drop_front(numProcessed), in __anon19d834b51002()
714 ivStorage.append(lbs.begin(), std::next(lbs.begin(), numProcessed));
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()
800 ValueRange tileSizes, AffineMap map, ValueRange lbs, in makeTiledShape() argument
829 auto offset = applyMapToValues(builder, loc, m, lbs).front(); in makeTiledShape()
1017 SmallVector<Value> lbs = computeTileOffsets(b, loc, ivs, tileSizes); in makeTiledShapes() local
1043 tiledShapes.push_back(makeTiledShape(b, loc, shapedOp, tileSizes, map, lbs, in makeTiledShapes()