Searched refs:reductionDimPos (Results 1 – 1 of 1) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | SplitReduction.cpp | 246 unsigned reductionDimPos, in scaleReductionDim() argument
248 auto reductionDim = getAffineDimExpr(reductionDimPos, op.getContext()); in scaleReductionDim()
261 unsigned reductionDimPos, int64_t size) { in insertParallelDim() argument
262 auto reductionDim = getAffineDimExpr(reductionDimPos, op.getContext()); in insertParallelDim()
289 unsigned reductionDimPos = dims[0]; in splitReductionByScaling() local
291 int64_t reductionDimSize = loopRanges[reductionDimPos]; in splitReductionByScaling()
364 auto redDim = getAffineDimExpr(reductionDimPos, context); in splitReductionByScaling()
365 auto redDimP1 = getAffineDimExpr(reductionDimPos + 1, context); in splitReductionByScaling()
372 newMaps.push_back(insertParallelDim(op, *o, reductionDimPos, in splitReductionByScaling()
389 iteratorTypes.insert(iteratorTypes.begin() + reductionDimPos, in splitReductionByScaling()
[all …]
|