Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparseTensorRewriting.cpp151 SmallVector<AffineMap> fusedIndexMaps = prod.getIndexingMapsArray(); in matchAndRewrite() local
153 fusedIndexMaps.push_back(fusedIndexMaps.back()); // mimic other in matchAndRewrite()
157 rewriter.getAffineMapArrayAttr(fusedIndexMaps), prod.iterator_types(), in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DElementwiseOpFusion.cpp264 SmallVector<AffineMap> fusedIndexMaps; in fuseElementwiseOpsImpl() local
267 fusedIndexMaps.reserve(producer->getNumOperands() + in fuseElementwiseOpsImpl()
277 fusedIndexMaps.push_back(consumer.getTiedIndexingMap(opOperand)); in fuseElementwiseOpsImpl()
289 fusedIndexMaps.push_back(map); in fuseElementwiseOpsImpl()
300 fusedIndexMaps.push_back(map); in fuseElementwiseOpsImpl()
307 fusedIndexMaps.push_back(consumer.getTiedIndexingMap(opOperand)); in fuseElementwiseOpsImpl()
311 fusedIndexMaps.push_back(consumer.getTiedIndexingMap(opOperand)); in fuseElementwiseOpsImpl()
1534 SmallVector<AffineMap> fusedIndexMaps; in matchAndRewrite() local
1537 fusedIndexMaps.reserve(genericOp.getNumInputsAndOutputs()); in matchAndRewrite()
1552 if (!inversePermutation(concatAffineMaps(fusedIndexMaps))) { in matchAndRewrite()
[all …]