Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DSplitReduction.cpp89 unsigned insertSplitDimension = control.second; in splitReduction() local
101 insertSplitDimension >= loopRanges.size()) in splitReduction()
164 if (idx == insertSplitDimension) { in splitReduction()
169 unsigned oldDim = idx < insertSplitDimension ? idx : idx - 1; in splitReduction()
195 if (insertSplitDimension == it.index()) in splitReduction()
215 if (insertSplitDimension == i) { in splitReduction()
280 unsigned insertSplitDimension = control.second; in splitReductionByScaling() local
294 insertSplitDimension >= loopRanges.size()) in splitReductionByScaling()
338 reductionDimSize / splitFactor, insertSplitDimension); in splitReductionByScaling()
416 map, map.dropResult(insertSplitDimension)}; in splitReductionByScaling()
[all …]