Lines Matching refs:indexingMap
585 getIndexingMapInExpandedOp(OpBuilder &builder, AffineMap indexingMap, in getIndexingMapInExpandedOp() argument
588 for (AffineExpr expr : indexingMap.getResults()) { in getIndexingMapInExpandedOp()
597 indexingMap.getNumSymbols(), newExprs, in getIndexingMapInExpandedOp()
604 AffineMap indexingMap, in getExpandedType() argument
607 for (AffineExpr expr : indexingMap.getResults()) { in getExpandedType()
622 getReassociationForExpansion(AffineMap indexingMap, in getReassociationForExpansion() argument
626 for (AffineExpr expr : indexingMap.getResults()) { in getReassociationForExpansion()
725 AffineMap indexingMap = genericOp.getTiedIndexingMap(opOperand); in fuseWithReshapeByExpansion() local
728 getExpandedType(opOperandType, indexingMap, expansionInfo); in fuseWithReshapeByExpansion()
732 getReassociationForExpansion(indexingMap, expansionInfo); in fuseWithReshapeByExpansion()
753 AffineMap indexingMap = genericOp.getTiedIndexingMap(opOperand); in fuseWithReshapeByExpansion() local
756 getExpandedType(opOperandType, indexingMap, expansionInfo); in fuseWithReshapeByExpansion()
759 getReassociationForExpansion(indexingMap, expansionInfo); in fuseWithReshapeByExpansion()
898 getDomainReassociation(AffineMap indexingMap, in getDomainReassociation() argument
900 assert(indexingMap.isProjectedPermutation() && in getDomainReassociation()
905 return indexingMap.getResults()[pos] in getDomainReassociation()
917 static bool isDimSequencePreserved(AffineMap indexingMap, in isDimSequencePreserved() argument
921 assert(indexingMap.isProjectedPermutation() && in isDimSequencePreserved()
928 for (const auto &expr : enumerate(indexingMap.getResults())) { in isDimSequencePreserved()
932 if (expr.index() + dimSequence.size() > indexingMap.getNumResults()) in isDimSequencePreserved()
937 indexingMap.getResult(expr.index() + dimInSequence.index()) in isDimSequencePreserved()
1026 AffineMap indexingMap = genericOp.getTiedIndexingMap(fusableOperand); in getCollapsableIterationSpaceDims() local
1037 getDomainReassociation(indexingMap, foldedRangeDims); in getCollapsableIterationSpaceDims()
1089 [&](AffineMap indexingMap) { in getCollapsableIterationSpaceDims() argument
1090 return !isDimSequencePreserved(indexingMap, in getCollapsableIterationSpaceDims()
1219 getCollapsedOpIndexingMap(AffineMap indexingMap, in getCollapsedOpIndexingMap() argument
1221 MLIRContext *context = indexingMap.getContext(); in getCollapsedOpIndexingMap()
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()
1254 indexingMap.getResult(counter).cast<AffineDimExpr>().getPosition(); in getOperandReassociation()
1275 AffineMap indexingMap = genericOp.getTiedIndexingMap(opOperand); in getCollapsedOpOperand() local
1277 getOperandReassociation(indexingMap, collapsingInfo); in getCollapsedOpOperand()
1282 if (operandReassociation.size() == indexingMap.getNumResults()) in getCollapsedOpOperand()
1419 AffineMap indexingMap = in collapseGenericOpIterationDims() local
1422 getOperandReassociation(indexingMap, collapsingInfo); in collapseGenericOpIterationDims()