Lines Matching refs:flatExprs
891 AffineExpr mlir::getAffineExprFromFlatForm(ArrayRef<int64_t> flatExprs, in getAffineExprFromFlatForm() argument
897 assert(flatExprs.size() - numDims - numSymbols - 1 == localExprs.size() && in getAffineExprFromFlatForm()
903 if (flatExprs[j] == 0) in getAffineExprFromFlatForm()
907 expr = expr + id * flatExprs[j]; in getAffineExprFromFlatForm()
911 for (unsigned j = numDims + numSymbols, e = flatExprs.size() - 1; j < e; in getAffineExprFromFlatForm()
913 if (flatExprs[j] == 0) in getAffineExprFromFlatForm()
915 auto term = localExprs[j - numDims - numSymbols] * flatExprs[j]; in getAffineExprFromFlatForm()
920 int64_t constTerm = flatExprs[flatExprs.size() - 1]; in getAffineExprFromFlatForm()
934 static AffineExpr getSemiAffineExprFromFlatForm(ArrayRef<int64_t> flatExprs, in getSemiAffineExprFromFlatForm() argument
939 assert(!flatExprs.empty() && "flatExprs cannot be empty"); in getSemiAffineExprFromFlatForm()
942 assert(flatExprs.size() - numDims - numSymbols - 1 == localExprs.size() && in getSemiAffineExprFromFlatForm()
991 if (flatExprs[j] == 0) in getSemiAffineExprFromFlatForm()
998 addEntry(indexEntry, flatExprs[j], getAffineDimExpr(j, context)); in getSemiAffineExprFromFlatForm()
1001 if (flatExprs[j] == 0) in getSemiAffineExprFromFlatForm()
1009 addEntry(indexEntry, flatExprs[j], 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()
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()
1095 int64_t constTerm = flatExprs.back(); in getSemiAffineExprFromFlatForm()