Lines Matching defs:ubs
131 SmallVectorImpl<Value> &ubs,
136 ubs.emplace_back(getValueOrCreateConstantIndexOp(builder, loc, range.size));
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());
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),
434 ubs.drop_front(numProcessed), steps.drop_front(numProcessed),
444 loc, lbs.take_front(numProcessed), ubs.take_front(numProcessed),
450 ubs.drop_front(numProcessed), steps.drop_front(numProcessed),
459 Value cond = ab.slt(lbs[0], ubs[0]);
461 cond = ab._and(cond, ab.slt(lbs[i], ubs[i]));
465 ubs.drop_front(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,
557 ArrayRef<OpFoldResult> ubs,
562 ubs, subShapeSizes, omitPartialTileCheck);
569 ArrayRef<OpFoldResult> lbs, ArrayRef<OpFoldResult> ubs,
662 llvm::to_vector(llvm::map_range(ubs, [&](OpFoldResult ub) {