Searched refs:newMaps (Results 1 – 1 of 1) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | SplitReduction.cpp | 117 SmallVector<AffineMap> newMaps; in splitReduction() local
141 newMaps.push_back( in splitReduction()
191 newMaps.push_back(AffineMap::get(oldOutputMap.getNumDims() + 1, 0, outputExpr, in splitReduction()
203 ValueRange({identityTensor}), newMaps, newIteratorTypes); in splitReduction()
358 SmallVector<AffineMap> newMaps; in splitReductionByScaling() local
359 newMaps.reserve(op.getNumInputsAndOutputs() + 1); in splitReductionByScaling()
361 newMaps.push_back(scaleReductionDim(op, *o, reductionDimPos, splitFactor)); in splitReductionByScaling()
366 newMaps.push_back(AffineMap::get(nDims, 0, {redDim, redDimP1}, context)); in splitReductionByScaling()
372 newMaps.push_back(insertParallelDim(op, *o, reductionDimPos, in splitReductionByScaling()
393 newOutputs, newMaps, iteratorTypes); in splitReductionByScaling()
|