Lines Matching refs:AffineMap
36 AffineMap map, in createFoldedComposedAffineApply()
45 AffineMap map, in applyMapToValues()
54 AffineMap map = AffineMap::get(numDims, numSym, expr); in applyMapToValues()
72 AffineMap map = getLoopsToShapesMap(); in createLoopRanges()
156 ArrayRef<AffineMap> indexingMaps, ArrayRef<StringRef> iteratorTypes, in build()
181 ValueRange outputBuffers, ArrayRef<AffineMap> indexingMaps, in build()
190 ValueRange outputBuffers, ArrayRef<AffineMap> indexingMaps, in build()
201 ArrayRef<AffineMap> indexingMaps, ArrayRef<StringRef> iteratorTypes, in build()
211 ArrayRef<AffineMap> indexingMaps, ArrayRef<StringRef> iteratorTypes, in build()
240 ValueRange outputBuffers, ArrayRef<AffineMap> indexingMaps, in build()
250 ValueRange outputBuffers, ArrayRef<AffineMap> indexingMaps, in build()
261 ArrayRef<AffineMap> indexingMaps, ArrayRef<StringRef> iteratorTypes, in build()
519 SmallVector<AffineMap, 4> indexingMaps; in verifyGenericOp()
573 static ArrayAttr collapseReassociationMaps(ArrayRef<AffineMap> mapsProducer, in collapseReassociationMaps()
574 ArrayRef<AffineMap> mapsConsumer, in collapseReassociationMaps()
588 for (AffineMap rhs : mapsConsumer) { in collapseReassociationMaps()
596 reassociationMaps.push_back(AffineMapAttr::get(AffineMap::get( in collapseReassociationMaps()
669 static bool isReassociationValid(ArrayRef<AffineMap> reassociation, in isReassociationValid()
728 ArrayRef<AffineMap> reassociation) { in computeReshapeCollapsedType()
745 for (AffineMap m : reassociation) { in computeReshapeCollapsedType()
789 static SmallVector<AffineMap, 4> getAffineMaps(ArrayAttr attrs) { in getAffineMaps()
808 static SmallVector<AffineMap, 4>
813 SmallVector<AffineMap, 4> maps; in getSymbolLessAffineMaps()
817 maps.push_back(AffineMap::get(maxDim + 1, 0, exprs, exprs[0].getContext())); in getSymbolLessAffineMaps()
930 ArrayRef<AffineMap> reassociation) { in computeTensorReshapeCollapsedType()
939 for (AffineMap m : reassociation) { in computeTensorReshapeCollapsedType()
1340 AffineMap mlir::linalg::extractOrIdentityMap(Optional<AffineMap> maybeMap, in extractOrIdentityMap()
1346 return AffineMap::get(context); in extractOrIdentityMap()
1347 return AffineMap::getMultiDimIdentityMap(rank, context); in extractOrIdentityMap()
1744 llvm::SmallDenseMap<std::pair<Value, AffineMap>, int> canonicalInput; in matchAndRewrite()
1751 AffineMap indexingMap = linalgOp.getInputIndexingMap(i); in matchAndRewrite()
1779 SmallVector<AffineMap, 6> newIndexingMaps; in matchAndRewrite()