Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DInlineScalarOperands.cpp36 SmallVector<AffineMap> newIndexingMaps; in matchAndRewrite() local
43 newIndexingMaps.emplace_back(map); in matchAndRewrite()
52 newIndexingMaps.emplace_back(genericOp.getTiedIndexingMap(opOperand)); in matchAndRewrite()
58 newIndexingMaps, in matchAndRewrite()
H A DDropUnitDims.cpp124 SmallVector<AffineMap, 4> newIndexingMaps; in replaceUnitDims() local
125 newIndexingMaps.reserve(indexingMaps.size()); in replaceUnitDims()
130 newIndexingMaps.push_back(simplifyAffineMap( in replaceUnitDims()
138 if (!inversePermutation(concatAffineMaps(newIndexingMaps))) in replaceUnitDims()
142 newIndexingMaps, [](AffineMap map) -> Attribute { in replaceUnitDims()
369 SmallVector<AffineMap> newIndexingMaps; in matchAndRewrite() local
377 newIndexingMaps.push_back(replacementInfo->indexMap); in matchAndRewrite()
386 newIndexingMaps.push_back(genericOp.getTiedIndexingMap(opOperand)); in matchAndRewrite()
401 !inversePermutation(concatAffineMaps(newIndexingMaps))) in matchAndRewrite()
433 loc, resultTypes, newInputs, newOutputs, newIndexingMaps, in matchAndRewrite()
H A DInterchange.cpp69 SmallVector<AffineMap> newIndexingMaps; in interchangeGenericOp() local
74 newIndexingMaps.push_back(m); in interchangeGenericOp()
77 rewriter.getAffineMapArrayAttr(newIndexingMaps)); in interchangeGenericOp()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorDropLeadUnitDim.cpp305 SmallVector<AffineMap> newIndexingMaps; in matchAndRewrite() local
382 newIndexingMaps.push_back(AffineMap::get(map.getNumDims() - 1, 0, results, in matchAndRewrite()
394 rewriter.getAffineMapArrayAttr(newIndexingMaps), in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/
H A DLinalgOps.cpp867 SmallVector<AffineMap> newIndexingMaps; in matchAndRewrite() local
872 newIndexingMaps); in matchAndRewrite()
877 newOutputOperands, newIndexingMaps); in matchAndRewrite()
893 rewriter.getAffineMapArrayAttr(newIndexingMaps), in matchAndRewrite()
930 SmallVector<AffineMap> &newIndexingMaps) const { in deduplicateInputOperands()
961 newIndexingMaps.push_back(indexingMap); in deduplicateInputOperands()
974 SmallVector<AffineMap> &newIndexingMaps) const { in deduplicateOutputOperands()
985 newIndexingMaps.push_back( in deduplicateOutputOperands()
1034 newIndexingMaps.push_back( in deduplicateOutputOperands()