Lines Matching refs:AffineExpr

216                                    AffineExpr expr, ValueRange dimValues,  in expandAffineExpr()
230 [numDims, &builder, loc, operands](AffineExpr expr) { in expandAffineMap()
459 AffineExpr mlir::substWithMin(AffineExpr e, AffineExpr dim, AffineExpr min, in substWithMin()
460 AffineExpr max, bool positivePath) { in substWithMin()
464 AffineExpr lhs = bin.getLHS(); in substWithMin()
465 AffineExpr rhs = bin.getRHS(); in substWithMin()
508 SmallVector<AffineExpr, 8> lbExprs; in normalizeAffineParallel()
509 SmallVector<AffineExpr, 8> ubExprs; in normalizeAffineParallel()
517 AffineExpr ubExpr = ranges.getResult(i).ceilDiv(step); in normalizeAffineParallel()
522 AffineExpr lbExpr = lbMap.getResult(i); in normalizeAffineParallel()
597 SmallVector<AffineExpr, 1> origLbExprs(origLbMap.getResults().begin(), in normalizeAffineFor()
599 SmallVector<AffineExpr, 2> origUbExprs(origUbMap.getResults().begin(), in normalizeAffineFor()
601 SmallVector<AffineExpr, 4> newUbExprs; in normalizeAffineFor()
643 AffineExpr origIVExpr = opBuilder.getAffineDimExpr(lb.getMap().getNumDims()); in normalizeAffineFor()
644 AffineExpr newIVExpr = origIVExpr * origLoopStep + origLbMap.getResult(0); in normalizeAffineFor()
1455 SmallVectorImpl<std::tuple<AffineExpr, unsigned, unsigned>> &tileSizePos) { in getTileSizePos() argument
1461 SmallVector<std::tuple<AffineExpr, AffineExpr, unsigned>, 4> floordivExprs; in getTileSizePos()
1463 for (AffineExpr expr : map.getResults()) { in getTileSizePos()
1474 tileSizePos = SmallVector<std::tuple<AffineExpr, unsigned, unsigned>>{}; in getTileSizePos()
1480 for (std::tuple<AffineExpr, AffineExpr, unsigned> fexpr : floordivExprs) { in getTileSizePos()
1481 AffineExpr floordivExprLHS = std::get<0>(fexpr); in getTileSizePos()
1482 AffineExpr floordivExprRHS = std::get<1>(fexpr); in getTileSizePos()
1494 for (AffineExpr expr : map.getResults()) { in getTileSizePos()
1497 expr.walk([&](AffineExpr e) { in getTileSizePos()
1532 tileSizePos = SmallVector<std::tuple<AffineExpr, unsigned, unsigned>>{}; in getTileSizePos()
1556 AffineExpr expr = layoutMap.getResults()[dim]; in isNormalizedMemRefDynamicDim()
1559 expr.walk([&inMemrefTypeDynDims, &isDynamicDim, &context](AffineExpr e) { in isNormalizedMemRefDynamicDim()
1572 static AffineExpr createDimSizeExprForTiledLayout(AffineExpr oldMapOutput, in createDimSizeExprForTiledLayout()
1577 AffineExpr newMapOutput; in createDimSizeExprForTiledLayout()
1653 SmallVector<std::tuple<AffineExpr, unsigned, unsigned>> tileSizePos; in createNewDynamicSizes()
1655 for (AffineExpr expr : map.getResults()) { in createNewDynamicSizes()
1665 AffineExpr newMapOutput = createDimSizeExprForTiledLayout(expr, pat); in createNewDynamicSizes()
1697 SmallVector<std::tuple<AffineExpr, unsigned, unsigned>> tileSizePos; in normalizeMemRef()
1754 SmallVector<std::tuple<AffineExpr, unsigned, unsigned>> tileSizePos; in normalizeMemRefType()