Lines Matching defs:indexingMap

593   // of the expanded op given the `indexingMap` of the fused operand/result of
696 /// `indexingMap` of the original operation.
698 getIndexingMapInExpandedOp(OpBuilder &builder, AffineMap indexingMap,
701   for (AffineExpr expr : indexingMap.getResults()) {
710                         indexingMap.getNumSymbols(), newExprs,
717                                         AffineMap indexingMap,
720   for (AffineExpr expr : indexingMap.getResults()) {
735 getReassociationForExpansion(AffineMap indexingMap,
739   for (AffineExpr expr : indexingMap.getResults()) {
875       AffineMap indexingMap = linalgOp.getMatchingIndexingMap(opOperand);
877           getExpandedType(opOperandType, indexingMap, expansionInfo);
881             getReassociationForExpansion(indexingMap, expansionInfo);
900     AffineMap indexingMap = linalgOp.getMatchingIndexingMap(&opOperand);
903         getExpandedType(opOperandType, indexingMap, expansionInfo);
906           getReassociationForExpansion(indexingMap, expansionInfo);
1142 /// For a given list of indices in the range of the `indexingMap` that are
1147 getDomainReassociation(AffineMap indexingMap,
1149   assert(indexingMap.isProjectedPermutation() &&
1154         return cast<AffineDimExpr>(indexingMap.getResults()[pos]).getPosition();
1162 /// `indexingMap`. `indexingMap` is expected to be a projected permutation.
1164 bool mlir::linalg::isDimSequencePreserved(AffineMap indexingMap,
1168   assert(indexingMap.isProjectedPermutation() &&
1175   for (const auto &expr : enumerate(indexingMap.getResults())) {
1179       if (expr.index() + dimSequence.size() > indexingMap.getNumResults())
1185                 indexingMap.getResult(expr.index() + dimInSequence.index()))
1278   AffineMap indexingMap = genericOp.getMatchingIndexingMap(fusableOperand);
1289         getDomainReassociation(indexingMap, foldedRangeDims);
1342                      [&](AffineMap indexingMap) {
1343                        return !isDimSequencePreserved(indexingMap,
1469 /// `indexingMap` of the original operation.
1471 getCollapsedOpIndexingMap(AffineMap indexingMap,
1473   MLIRContext *context = indexingMap.getContext();
1474   assert(indexingMap.isProjectedPermutation() &&
1479   for (auto expr : indexingMap.getResults()) {
1496 getOperandReassociation(AffineMap indexingMap,
1504   while (counter < indexingMap.getNumResults()) {
1506         cast<AffineDimExpr>(indexingMap.getResult(counter)).getPosition();
1527   AffineMap indexingMap = op.getMatchingIndexingMap(opOperand);
1529       getOperandReassociation(indexingMap, collapsingInfo);
1535   if (operandReassociation.size() == indexingMap.getNumResults())
1746       AffineMap indexingMap =
1749           getOperandReassociation(indexingMap, collapsingInfo);