Lines Matching refs:tripCounts

135                                       ArrayRef<Value> tripCounts) {  in delinearize()  argument
136 SmallVector<Value> coords(tripCounts.size()); in delinearize()
137 assert(!tripCounts.empty() && "tripCounts must be not empty"); in delinearize()
139 for (ssize_t i = tripCounts.size() - 1; i >= 0; --i) { in delinearize()
140 coords[i] = b.create<SignedRemIOp>(index, tripCounts[i]); in delinearize()
141 index = b.create<SignedDivIOp>(index, tripCounts[i]); in delinearize()
229 ArrayRef<Value> tripCounts = getArguments(op.getNumLoops()); in createParallelComputeFunction() local
233 Value tripCount = tripCounts[0]; in createParallelComputeFunction()
234 for (unsigned i = 1; i < tripCounts.size(); ++i) in createParallelComputeFunction()
235 tripCount = b.create<MulIOp>(tripCount, tripCounts[i]); in createParallelComputeFunction()
257 auto blockFirstCoord = delinearize(b, blockFirstIndex, tripCounts); in createParallelComputeFunction()
258 auto blockLastCoord = delinearize(b, blockLastIndex, tripCounts); in createParallelComputeFunction()
336 tripCounts[loopIdx + 1]); in createParallelComputeFunction()
496 const SmallVector<Value> &tripCounts) { in doAsyncDispatch() argument
516 operands.append(tripCounts); in doAsyncDispatch()
566 const SmallVector<Value> &tripCounts) { in doSequantialDispatch() argument
587 computeFuncOperands.append(tripCounts); in doSequantialDispatch()
638 SmallVector<Value> tripCounts(op.getNumLoops()); in matchAndRewrite() local
644 tripCounts[i] = b.create<SignedCeilDivIOp>(range, step); in matchAndRewrite()
649 Value tripCount = tripCounts[0]; in matchAndRewrite()
650 for (size_t i = 1; i < tripCounts.size(); ++i) in matchAndRewrite()
651 tripCount = b.create<MulIOp>(tripCount, tripCounts[i]); in matchAndRewrite()
709 blockCount, tripCounts); in matchAndRewrite()
712 blockCount, tripCounts); in matchAndRewrite()