Lines Matching defs:tripCounts

142   ArrayRef<BlockArgument> tripCounts();
152   SmallVector<IntegerAttr> tripCounts;
169 ArrayRef<BlockArgument> ParallelComputeFunctionArgs::tripCounts() {
196                                       ArrayRef<Value> tripCounts) {
197   SmallVector<Value> coords(tripCounts.size());
198   assert(!tripCounts.empty() && "tripCounts must be not empty");
200   for (ssize_t i = tripCounts.size() - 1; i >= 0; --i) {
201     coords[i] = b.create<arith::RemSIOp>(index, tripCounts[i]);
202     index = b.create<arith::DivSIOp>(index, tripCounts[i]);
300   auto tripCounts = values(args.tripCounts(), bounds.tripCounts);
311   Value tripCount = tripCounts[0];
312   for (unsigned i = 1; i < tripCounts.size(); ++i)
313     tripCount = b.create<arith::MulIOp>(tripCount, tripCounts[i]);
326   auto blockFirstCoord = delinearize(b, blockFirstIndex, tripCounts);
327   auto blockLastCoord = delinearize(b, blockLastIndex, tripCounts);
401           b.create<scf::ForOp>(c0, tripCounts[loopIdx + 1], c1, ValueRange(),
412                                               tripCounts[loopIdx + 1]);
580                             const SmallVector<Value> &tripCounts) {
594     operands.append(tripCounts);
652                      const SmallVector<Value> &tripCounts) {
673     computeFuncOperands.append(tripCounts);
735   SmallVector<Value> tripCounts(op.getNumLoops());
741     tripCounts[i] = b.createOrFold<arith::CeilDivSIOp>(range, step);
746   Value tripCount = tripCounts[0];
747   for (size_t i = 1; i < tripCounts.size(); ++i)
748     tripCount = b.create<arith::MulIOp>(tripCount, tripCounts[i]);
772         integerConstants(tripCounts),
794     numIterations.back() = getInt(staticBounds.tripCounts.back());
797       int64_t tripCount = getInt(staticBounds.tripCounts[i]);
878       doDispatch(b, rewriter, compute, op, blockSize, blockCount, tripCounts);
895                  tripCounts);