Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DElementwiseOpFusion.cpp898 getDomainReassociation(AffineMap indexingMap, in getDomainReassociation() argument
900 assert(indexingMap.isProjectedPermutation() && in getDomainReassociation()
905 return indexingMap.getResults()[pos] in getDomainReassociation()
921 assert(indexingMap.isProjectedPermutation() && in isDimSequencePreserved()
1089 [&](AffineMap indexingMap) { in getCollapsableIterationSpaceDims() argument
1219 getCollapsedOpIndexingMap(AffineMap indexingMap, in getCollapsedOpIndexingMap() argument
1222 assert(indexingMap.isProjectedPermutation() && in getCollapsedOpIndexingMap()
1227 for (auto expr : indexingMap.getResults()) { in getCollapsedOpIndexingMap()
1244 getOperandReassociation(AffineMap indexingMap, in getOperandReassociation() argument
1252 while (counter < indexingMap.getNumResults()) { in getOperandReassociation()
[all …]
H A DBubbleUpExtractSlice.cpp84 AffineMap indexingMap = linalgOp.getTiedIndexingMap(outOperand); in matchAndRewrite() local
85 if (!indexingMap.isProjectedPermutation()) { in matchAndRewrite()
114 for (auto const &result : enumerate(indexingMap.getResults())) { in matchAndRewrite()
H A DTilingInterfaceImpl.cpp148 AffineMap indexingMap = in generateResultTileValue() local
150 if (!indexingMap.isProjectedPermutation()) { in generateResultTileValue()
160 if (!indexingMap.isPermutation()) { in generateResultTileValue()
168 for (const auto &resultExpr : llvm::enumerate(indexingMap.getResults())) { in generateResultTileValue()
H A DDropUnitDims.cpp242 AffineMap indexingMap = genericOp.getTiedIndexingMap(opOperand); in replaceUnitExtents() local
244 ArrayRef<AffineExpr> exprs = indexingMap.getResults(); in replaceUnitExtents()
295 AffineMap::get(indexingMap.getNumDims(), in replaceUnitExtents()
296 indexingMap.getNumSymbols(), in replaceUnitExtents()
H A DDecomposeLinalgOps.cpp171 AffineMap indexingMap = in createPeeledGenericOp() local
173 if (indexingMap.isIdentity()) in createPeeledGenericOp()
H A DFusionOnTensors.cpp42 AffineMap indexingMap = consumerOp.getTiedIndexingMap(consumerOperand); in getTiledSliceDims() local
46 for (const auto &en : enumerate(indexingMap.getResults())) { in getTiledSliceDims()
H A DTransforms.cpp172 AffineMap indexingMap = opToPad.getTiedIndexingMap(opOperand); in padOperandToSmallestStaticBoundingBox() local
178 for (const auto &en : enumerate(indexingMap.getResults())) in padOperandToSmallestStaticBoundingBox()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/
H A DLinalgInterfaces.cpp656 AffineMap indexingMap = linalgOp.getTiedIndexingMap(opOperand); in verifyStructuredOpInterface() local
659 if (indexingMap.getNumSymbols() != 0) in verifyStructuredOpInterface()
665 if (indexingMap.getNumDims() != numLoops) in verifyStructuredOpInterface()
671 if (indexingMap.getNumResults() != rank) in verifyStructuredOpInterface()
675 << indexingMap.getNumResults() << ")"; in verifyStructuredOpInterface()
703 AffineMap indexingMap = linalgOp.getTiedIndexingMap(opOperand); in verifyStructuredOpInterface() local
704 for (AffineExpr expr : indexingMap.getResults()) { in verifyStructuredOpInterface()
764 indexingMap.compose(startLoopRangeValues); in verifyStructuredOpInterface()
766 indexingMap.compose(endLoopRangeValues); in verifyStructuredOpInterface()
789 os << indexingMap; in verifyStructuredOpInterface()
[all …]
H A DLinalgOps.cpp946 AffineMap indexingMap = in deduplicateInputOperands() local
949 std::make_pair(inputOpOperand.value()->get(), indexingMap)); in deduplicateInputOperands()
958 dedupedInputs[{inputOpOperand.value()->get(), indexingMap}] = in deduplicateInputOperands()
961 newIndexingMaps.push_back(indexingMap); in deduplicateInputOperands()
998 AffineMap indexingMap = in deduplicateOutputOperands() local
1001 std::make_tuple(outputOpOperand.value()->get(), indexingMap, in deduplicateOutputOperands()
/llvm-project-15.0.7/mlir/tools/mlir-linalg-ods-gen/
H A Dmlir-linalg-ods-yaml-gen.cpp904 for (auto &indexingMap : *staticMaps) { in generateNamedGenericOpDefns()
908 generateCppExpression(indexingMap, "context"))); in generateNamedGenericOpDefns()