| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | BubbleUpExtractSlice.cpp | 84 AffineMap indexingMap = linalgOp.getMatchingIndexingMap(outOperand); in matchAndRewrite() local
|
| H A D | ElementwiseOpFusion.cpp | 652 getIndexingMapInExpandedOp(OpBuilder & builder,AffineMap indexingMap,const ExpansionInfo & expansionInfo) getIndexingMapInExpandedOp() argument
671 getExpandedType(RankedTensorType originalType,AffineMap indexingMap,const ExpansionInfo & expansionInfo) getExpandedType() argument
689 getReassociationForExpansion(AffineMap indexingMap,const ExpansionInfo & expansionInfo) getReassociationForExpansion() argument
829 AffineMap indexingMap = linalgOp.getMatchingIndexingMap(opOperand); fuseWithReshapeByExpansion() local
854 AffineMap indexingMap = linalgOp.getMatchingIndexingMap(&opOperand); fuseWithReshapeByExpansion() local
1101 getDomainReassociation(AffineMap indexingMap,ReassociationIndicesRef rangeReassociation) getDomainReassociation() argument
1118 isDimSequencePreserved(AffineMap indexingMap,ReassociationIndicesRef dimSequence) isDimSequencePreserved() argument
1232 AffineMap indexingMap = genericOp.getMatchingIndexingMap(fusableOperand); getCollapsableIterationSpaceDims() local
1296 __anona62ab6fa1602(AffineMap indexingMap) getCollapsableIterationSpaceDims() argument
1425 getCollapsedOpIndexingMap(AffineMap indexingMap,const CollapsingInfo & collapsingInfo) getCollapsedOpIndexingMap() argument
1450 getOperandReassociation(AffineMap indexingMap,const CollapsingInfo & collapsingInfo) getOperandReassociation() argument
1481 AffineMap indexingMap = op.getMatchingIndexingMap(opOperand); getCollapsedOpOperand() local
1700 AffineMap indexingMap = collapseOpIterationDims() local
[all...] |
| H A D | RuntimeOpVerification.cpp | 51 AffineMap indexingMap = linalgOp.getMatchingIndexingMap(&opOperand); in generateRuntimeVerification() local
|
| H A D | TilingInterfaceImpl.cpp | 33 AffineMap indexingMap, in getIndicesForAccess() argument
177 AffineMap indexingMap = getIterationDomainTileFromOperandTile() local
229 AffineMap indexingMap = getIterationDomainTileFromResultTile() local
[all...] |
| H A D | EraseUnusedOperandsAndResults.cpp | 153 AffineMap indexingMap = genericOp.getMatchingIndexingMap(inputOpOperand); in deduplicateInputOperands() local
204 AffineMap indexingMap = in deduplicateOutputOperands() local
|
| H A D | Padding.cpp | 32 AffineMap indexingMap = opToPad.getMatchingIndexingMap(opOperand); in computePaddedShape() local
|
| H A D | DecomposeLinalgOps.cpp | 190 AffineMap indexingMap = rewriter.getMultiDimIdentityMap(domain.size()); in createPeeledGenericOp() local
|
| H A D | DataLayoutPropagation.cpp | 69 AffineMap indexingMap = genericOp.getMatchingIndexingMap(opOperand); in getPackingInfoFromOperand() local
287 auto indexingMap = AffineMap::get(numLoops, 0, exprs, b.getContext()); in getOrCreatePackedViewOfOperand() local
|
| H A D | DropUnitDims.cpp | 347 AffineMap indexingMap = genericOp.getMatchingIndexingMap(opOperand); dropUnitExtentFromOperandMetadata() local
468 auto indexingMap = genericOp.getMatchingIndexingMap(&opOperand); dropUnitDims() local
|
| H A D | Vectorization.cpp | 1311 AffineMap indexingMap = linalgOp.getMatchingIndexingMap(opOperand); vectorizeAsLinalgGeneric() local
1729 AffineMap indexingMap = op.getMatchingIndexingMap(&opOperand); reductionPreconditions() local
|
| /llvm-project/mlir/lib/Dialect/Linalg/TransformOps/ |
| H A D | LinalgMatchOps.cpp | 447 AffineMap indexingMap = in matchOperation() local
538 AffineMap indexingMap = in matchOperation() local
|
| /llvm-project/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgInterfaces.cpp | 280 findPermutationsIndexingOperand(AffineMap indexingMap,ArrayRef<utils::IteratorType> iterators,utils::IteratorType iter) findPermutationsIndexingOperand() argument
1157 AffineMap indexingMap = linalgOp.getMatchingIndexingMap(&opOperand); verifyStructuredOpInterface() local
1195 AffineMap indexingMap = linalgOp.getMatchingIndexingMap(&opOperand); verifyStructuredOpInterface() local
[all...] |