Lines Matching refs:AffineMap
38 static AffineMap getIndexingMapOfProducerOperandsInCoordinatesOfFusedOp( in getIndexingMapOfProducerOperandsInCoordinatesOfFusedOp()
39 OpOperand *producerOpOperand, AffineMap producerResultIndexMap, in getIndexingMapOfProducerOperandsInCoordinatesOfFusedOp()
40 AffineMap fusedConsumerArgIndexMap) { in getIndexingMapOfProducerOperandsInCoordinatesOfFusedOp()
49 AffineMap invProducerResultIndexMap = in getIndexingMapOfProducerOperandsInCoordinatesOfFusedOp()
56 AffineMap argMap = producer.getTiedIndexingMap(producerOpOperand); in getIndexingMapOfProducerOperandsInCoordinatesOfFusedOp()
60 AffineMap t1 = argMap.compose(invProducerResultIndexMap); in getIndexingMapOfProducerOperandsInCoordinatesOfFusedOp()
86 AffineMap consumerIndexMap = consumer.getTiedIndexingMap(consumerOpOperand); in areElementwiseOpsFusable()
96 AffineMap producerResultIndexMap = in areElementwiseOpsFusable()
108 auto addToCoveredDims = [&](AffineMap map) { in areElementwiseOpsFusable()
119 AffineMap operandMap = std::get<1>(pair); in areElementwiseOpsFusable()
124 AffineMap newIndexingMap = in areElementwiseOpsFusable()
140 AffineMap consumerToProducerLoopsMap, in generateFusedElementwiseOpRegion()
264 SmallVector<AffineMap> fusedIndexMaps; in fuseElementwiseOpsImpl()
281 AffineMap producerResultIndexMap = in fuseElementwiseOpsImpl()
286 AffineMap map = getIndexingMapOfProducerOperandsInCoordinatesOfFusedOp( in fuseElementwiseOpsImpl()
297 AffineMap map = getIndexingMapOfProducerOperandsInCoordinatesOfFusedOp( in fuseElementwiseOpsImpl()
336 AffineMap consumerResultIndexMap = in fuseElementwiseOpsImpl()
339 AffineMap invProducerResultIndexMap = in fuseElementwiseOpsImpl()
344 AffineMap consumerToProducerLoopsMap = in fuseElementwiseOpsImpl()
484 ArrayRef<AffineMap> reassociationMaps,
513 ArrayRef<AffineMap> reassociationMaps, in compute()
519 AffineMap fusedIndexMap = linalgOp.getTiedIndexingMap(fusableOpOperand); in compute()
532 AffineMap foldedDims = reassociationMaps[resultExpr.index()]; in compute()
584 static AffineMap
585 getIndexingMapInExpandedOp(OpBuilder &builder, AffineMap indexingMap, in getIndexingMapInExpandedOp()
596 return AffineMap::get(expansionInfo.getExpandedOpNumDims(), in getIndexingMapInExpandedOp()
604 AffineMap indexingMap, in getExpandedType()
622 getReassociationForExpansion(AffineMap indexingMap, in getReassociationForExpansion()
711 SmallVector<AffineMap, 4> expandedOpIndexingMaps = llvm::to_vector<4>( in fuseWithReshapeByExpansion()
712 llvm::map_range(genericOp.getIndexingMapsArray(), [&](AffineMap m) { in fuseWithReshapeByExpansion()
725 AffineMap indexingMap = genericOp.getTiedIndexingMap(opOperand); in fuseWithReshapeByExpansion()
753 AffineMap indexingMap = genericOp.getTiedIndexingMap(opOperand); in fuseWithReshapeByExpansion()
898 getDomainReassociation(AffineMap indexingMap, in getDomainReassociation()
917 static bool isDimSequencePreserved(AffineMap indexingMap, in isDimSequencePreserved()
1011 if (!llvm::all_of(genericOp.getIndexingMapsArray(), [](AffineMap map) { in getCollapsableIterationSpaceDims()
1026 AffineMap indexingMap = genericOp.getTiedIndexingMap(fusableOperand); in getCollapsableIterationSpaceDims()
1089 [&](AffineMap indexingMap) { in getCollapsableIterationSpaceDims()
1218 static AffineMap
1219 getCollapsedOpIndexingMap(AffineMap indexingMap, in getCollapsedOpIndexingMap()
1237 return AffineMap::get(collapsingInfo.getCollapsedOpIterationRank(), 0, in getCollapsedOpIndexingMap()
1244 getOperandReassociation(AffineMap indexingMap, in getOperandReassociation()
1275 AffineMap indexingMap = genericOp.getTiedIndexingMap(opOperand); in getCollapsedOpOperand()
1355 llvm::map_range(genericOp.getIndexingMapsArray(), [&](AffineMap map) { in collapseGenericOpIterationDims()
1419 AffineMap indexingMap = in collapseGenericOpIterationDims()
1534 SmallVector<AffineMap> fusedIndexMaps; in matchAndRewrite()