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