Lines Matching refs:getAffineDimExpr
131 exprs.push_back(b.getAffineDimExpr(insertSplitDimension)); in splitReduction()
133 b.getAffineDimExpr(dim < insertSplitDimension ? dim : dim + 1)); in splitReduction()
138 b.getAffineDimExpr(dim < insertSplitDimension ? dim : dim + 1)); in splitReduction()
166 outputExpr.push_back(b.getAffineDimExpr(insertSplitDimension)); in splitReduction()
173 b.getAffineDimExpr(dim < insertSplitDimension ? dim : dim + 1)); in splitReduction()
218 exprs.push_back(b.getAffineDimExpr(i)); in splitReduction()
248 auto reductionDim = getAffineDimExpr(reductionDimPos, op.getContext()); in scaleReductionDim()
249 auto reductionDimP1 = getAffineDimExpr(reductionDimPos + 1, op.getContext()); in scaleReductionDim()
262 auto reductionDim = getAffineDimExpr(reductionDimPos, op.getContext()); in insertParallelDim()
364 auto redDim = getAffineDimExpr(reductionDimPos, context); in splitReductionByScaling()
365 auto redDimP1 = getAffineDimExpr(reductionDimPos + 1, context); in splitReductionByScaling()