Home
last modified time | relevance | path

Searched refs:getUnrollFactor (Results 1 – 2 of 2) sorted by relevance

/external/llvm-project/mlir/lib/Dialect/Affine/Transforms/
DLoopUnroll.cpp40 const std::function<unsigned(AffineForOp)> getUnrollFactor; member
42 LoopUnroll() : getUnrollFactor(nullptr) {} in LoopUnroll()
45 getUnrollFactor(other.getUnrollFactor) {} in LoopUnroll()
49 const std::function<unsigned(AffineForOp)> &getUnrollFactor = nullptr) in LoopUnroll()
50 : getUnrollFactor(getUnrollFactor) { in LoopUnroll()
106 for (unsigned i = 0; i < numRepetitions || getUnrollFactor; i++) { in runOnFunction()
124 if (getUnrollFactor) in runOnAffineForOp()
125 return loopUnrollByFactor(forOp, getUnrollFactor(forOp)); in runOnAffineForOp()
137 const std::function<unsigned(AffineForOp)> &getUnrollFactor) { in createLoopUnrollPass() argument
140 unrollUpToFactor, unrollFull, getUnrollFactor); in createLoopUnrollPass()
/external/llvm-project/mlir/include/mlir/Dialect/Affine/
DPasses.h66 const std::function<unsigned(AffineForOp)> &getUnrollFactor = nullptr);