Lines Matching defs:lbs

130                          ArrayRef<Range> ranges, SmallVectorImpl<Value> &lbs,
134 lbs.emplace_back(
290 SmallVector<Value, 4> lbs, ubs, steps;
291 unpackRanges(b, loc, loopRanges, lbs, ubs, steps);
293 b, loc, lbs, ubs, steps, iterArgInitValues,
331 SmallVector<Value, 4> lbs, ubs, steps;
332 unpackRanges(b, loc, loopRanges, lbs, ubs, steps);
343 affine::buildAffineLoopNest(b, loc, lbs, ubs, constantSteps,
373 OpBuilder &b, Location loc, ValueRange lbs, ValueRange ubs,
378 assert(lbs.size() == ubs.size());
379 assert(lbs.size() == steps.size());
380 assert(lbs.size() == iteratorTypes.size());
381 assert(procInfo.empty() || (lbs.size() == procInfo.size()));
394 b, loc, lbs.take_front(), ubs.take_front(), steps.take_front(),
398 b, loc, lbs.drop_front(), ubs.drop_front(), steps.drop_front(),
428 loc, lbs.take_front(numProcessed), ubs.take_front(numProcessed),
433 nestedBuilder, nestedLoc, lbs.drop_front(numProcessed),
444 loc, lbs.take_front(numProcessed), ubs.take_front(numProcessed),
449 nestedBuilder, nestedLoc, lbs.drop_front(numProcessed),
459 Value cond = ab.slt(lbs[0], ubs[0]);
461 cond = ab._and(cond, ab.slt(lbs[i], ubs[i]));
462 ivStorage.append(lbs.begin(), std::next(lbs.begin(), numProcessed));
464 generateParallelLoopNest(b, loc, lbs.drop_front(numProcessed),
476 ivStorage.append(lbs.begin(), std::next(lbs.begin(), numProcessed));
478 b, loc, lbs.drop_front(numProcessed), ubs.drop_front(numProcessed),
522 ValueRange lbs(lbsStorage), ubs(ubsStorage), steps(stepsStorage);
524 b, loc, lbs, ubs, steps, iteratorTypes, procInfo,
556 ArrayRef<OpFoldResult> lbs,
561 computeSliceParameters(builder, loc, valueToTile, tileSizes, map, lbs,
569 ArrayRef<OpFoldResult> lbs, ArrayRef<OpFoldResult> ubs,
599 // The offset of the slice is m(lbs) - m(0).
600 SmallVector<Attribute> zeros(lbs.size(), rewriter.getIndexAttr(0));
607 rewriter, loc, m.getResult(0) - mAtZeroInt, lbs);
764 SmallVector<OpFoldResult> lbs =
796 builder, loc, shapedOp, tileSizes, map, lbs, sizeBounds, subShapeSizes,