Lines Matching refs:numDims
108 AffineExpr AffineExpr::shiftDims(unsigned numDims, unsigned shift, in shiftDims() argument
113 for (unsigned idx = offset; idx < numDims; ++idx) in shiftDims()
892 unsigned numDims, in getAffineExprFromFlatForm() argument
897 assert(flatExprs.size() - numDims - numSymbols - 1 == localExprs.size() && in getAffineExprFromFlatForm()
902 for (unsigned j = 0; j < numDims + numSymbols; j++) { in getAffineExprFromFlatForm()
905 auto id = j < numDims ? getAffineDimExpr(j, context) in getAffineExprFromFlatForm()
906 : getAffineSymbolExpr(j - numDims, context); in getAffineExprFromFlatForm()
911 for (unsigned j = numDims + numSymbols, e = flatExprs.size() - 1; j < e; in getAffineExprFromFlatForm()
915 auto term = localExprs[j - numDims - numSymbols] * flatExprs[j]; in getAffineExprFromFlatForm()
935 unsigned numDims, in getSemiAffineExprFromFlatForm() argument
942 assert(flatExprs.size() - numDims - numSymbols - 1 == localExprs.size() && in getSemiAffineExprFromFlatForm()
990 for (unsigned j = 0; j < numDims; ++j) { in getSemiAffineExprFromFlatForm()
1000 for (unsigned j = numDims; j < numDims + numSymbols; ++j) { in getSemiAffineExprFromFlatForm()
1007 std::pair<unsigned, signed> indexEntry(j - numDims, in getSemiAffineExprFromFlatForm()
1008 std::max(numDims, numSymbols)); in getSemiAffineExprFromFlatForm()
1010 getAffineSymbolExpr(j - numDims, context)); in getSemiAffineExprFromFlatForm()
1015 SmallVector<bool, 4> addedToMap(flatExprs.size() - numDims - numSymbols - 1, in getSemiAffineExprFromFlatForm()
1024 if (flatExprs[numDims + numSymbols + it.index()] == 0) in getSemiAffineExprFromFlatForm()
1041 addEntry(indexEntry, flatExprs[numDims + numSymbols + it.index()], in getSemiAffineExprFromFlatForm()
1046 std::max(numDims, numSymbols)); in getSemiAffineExprFromFlatForm()
1047 addEntry(indexEntry, flatExprs[numDims + numSymbols + it.index()], in getSemiAffineExprFromFlatForm()
1059 addEntry(indexEntry, flatExprs[numDims + numSymbols + it.index()], expr); in getSemiAffineExprFromFlatForm()
1068 addEntry(indexEntry, flatExprs[numDims + numSymbols + it.index()], expr); in getSemiAffineExprFromFlatForm()
1084 for (unsigned j = numDims + numSymbols, e = flatExprs.size() - 1; j < e; in getSemiAffineExprFromFlatForm()
1088 if (flatExprs[j] == 0 || addedToMap[j - numDims - numSymbols]) in getSemiAffineExprFromFlatForm()
1090 auto term = localExprs[j - numDims - numSymbols] * flatExprs[j]; in getSemiAffineExprFromFlatForm()
1101 SimpleAffineExprFlattener::SimpleAffineExprFlattener(unsigned numDims, in SimpleAffineExprFlattener() argument
1103 : numDims(numDims), numSymbols(numSymbols), numLocals(0) { in SimpleAffineExprFlattener()
1123 AffineExpr a = getAffineExprFromFlatForm(lhs, numDims, numSymbols, in visitMulExpr()
1125 AffineExpr b = getAffineExprFromFlatForm(rhs, numDims, numSymbols, in visitMulExpr()
1174 lhs, numDims, numSymbols, localExprs, context); in visitModExpr()
1175 AffineExpr divisorExpr = getAffineExprFromFlatForm(rhs, numDims, numSymbols, in visitModExpr()
1215 floorDividend, numDims, numSymbols, localExprs, context); in visitModExpr()
1239 assert(expr.getPosition() < numDims && "Inconsistent number of dims"); in visitDimExpr()
1297 AffineExpr a = getAffineExprFromFlatForm(lhs, numDims, numSymbols, in visitDivExpr()
1299 AffineExpr b = getAffineExprFromFlatForm(rhs, numDims, numSymbols, in visitDivExpr()
1333 getAffineExprFromFlatForm(lhs, numDims, numSymbols, localExprs, context); in visitDivExpr()
1389 AffineExpr mlir::simplifyAffineExpr(AffineExpr expr, unsigned numDims, in simplifyAffineExpr() argument
1395 SimpleAffineExprFlattener flattener(numDims, numSymbols); in simplifyAffineExpr()
1399 expr == getAffineExprFromFlatForm(flattenedExpr, numDims, numSymbols, in simplifyAffineExpr()
1405 ? getAffineExprFromFlatForm(flattenedExpr, numDims, numSymbols, in simplifyAffineExpr()
1407 : getSemiAffineExprFromFlatForm(flattenedExpr, numDims, numSymbols, in simplifyAffineExpr()