Lines Matching refs:getNumLoops
218 inputs.reserve(2 + 4 * op.getNumLoops() + captures.size()); in getParallelComputeFunctionType()
227 for (unsigned i = 0; i < op.getNumLoops(); ++i) in getParallelComputeFunctionType()
233 for (unsigned i = 0; i < op.getNumLoops(); ++i) { in getParallelComputeFunctionType()
279 ParallelComputeFunctionArgs args = {op.getNumLoops(), func.getArguments()}; in createParallelComputeFunction()
334 SmallVector<Value> blockEndCoord(op.getNumLoops()); in createParallelComputeFunction()
349 SmallVector<Value> computeBlockInductionVars(op.getNumLoops()); in createParallelComputeFunction()
368 SmallVector<Value> isBlockFirstCoord(op.getNumLoops()); in createParallelComputeFunction()
369 SmallVector<Value> isBlockLastCoord(op.getNumLoops()); in createParallelComputeFunction()
397 if (loopIdx < op.getNumLoops() - 1) { in createParallelComputeFunction()
398 if (loopIdx + 1 >= op.getNumLoops() - numBlockAlignedInnerLoops) { in createParallelComputeFunction()
436 return {op.getNumLoops(), func, std::move(computeFuncType.captures)}; in createParallelComputeFunction()
735 SmallVector<Value> tripCounts(op.getNumLoops()); in matchAndRewrite()
736 for (size_t i = 0; i < op.getNumLoops(); ++i) { in matchAndRewrite()
793 SmallVector<int64_t> numIterations(op.getNumLoops()); in matchAndRewrite()
796 for (int i = op.getNumLoops() - 2; i >= 0; --i) { in matchAndRewrite()
891 numIterations[op.getNumLoops() - numUnrollableLoops]); in matchAndRewrite()
904 numIterations[op.getNumLoops() - numUnrollableLoops]); in matchAndRewrite()