Lines Matching refs:AffineExpr
35 IntegerAttr constantFold(AffineExpr expr) { in constantFold()
42 Optional<int64_t> constantFoldImpl(AffineExpr expr) { in constantFoldImpl()
77 Optional<int64_t> constantFoldBinExpr(AffineExpr expr, in constantFoldBinExpr()
126 AffineExpr expr = idxAndExpr.value(); in isMinorIdentityWithBroadcasting()
173 AffineExpr expr = idxAndExpr.value(); in isPermutationOfMinorIdentityWithBroadcasting()
209 SmallVector<AffineExpr, 4> affExprs; in getPermutationMap()
235 AffineMap::inferFromExprList(ArrayRef<ArrayRef<AffineExpr>> exprsList) { in inferFromExprList()
240 AffineMap::inferFromExprList(ArrayRef<SmallVector<AffineExpr, 4>> exprsList) { in inferFromExprList() argument
246 SmallVector<AffineExpr, 4> dimExprs; in getMultiDimIdentityMap()
258 ArrayRef<AffineExpr> results = getResults(); in isIdentity()
276 return llvm::all_of(getResults(), [](AffineExpr expr) { in isConstant()
307 ArrayRef<AffineExpr> AffineMap::getResults() const { in getResults()
311 AffineExpr AffineMap::getResult(unsigned idx) const { in getResult()
356 SmallVector<AffineExpr, 4> exprs; in partialConstantFold()
382 void AffineMap::walkExprs(llvm::function_ref<void(AffineExpr)> callback) const { in walkExprs()
392 AffineMap AffineMap::replaceDimsAndSymbols(ArrayRef<AffineExpr> dimReplacements, in replaceDimsAndSymbols()
393 ArrayRef<AffineExpr> symReplacements, in replaceDimsAndSymbols()
396 SmallVector<AffineExpr, 8> results; in replaceDimsAndSymbols()
407 AffineMap AffineMap::replace(AffineExpr expr, AffineExpr replacement, in replace()
410 SmallVector<AffineExpr, 4> newResults; in replace()
412 for (AffineExpr e : getResults()) in replace()
420 AffineMap AffineMap::replace(const DenseMap<AffineExpr, AffineExpr> &map, in replace() argument
423 SmallVector<AffineExpr, 4> newResults; in replace()
425 for (AffineExpr e : getResults()) in replace()
431 AffineMap::replace(const DenseMap<AffineExpr, AffineExpr> &map) const { in replace() argument
432 SmallVector<AffineExpr, 4> newResults; in replace()
434 for (AffineExpr e : getResults()) in replace()
445 SmallVector<AffineExpr, 8> newDims(numDims); in compose()
449 SmallVector<AffineExpr, 8> newSymbols(numSymbols - numSymbolsThisMap); in compose()
456 SmallVector<AffineExpr, 8> exprs; in compose()
465 SmallVector<AffineExpr, 4> exprs; in compose()
515 SmallVector<AffineExpr, 4> exprs; in getSubMap()
546 SmallVector<AffineExpr> dimReplacements; in compressDims()
555 SmallVector<AffineExpr> resultExprs; in compressDims()
571 SmallVector<AffineExpr> allExprs; in compressUnusedImpl()
584 ArrayRef<AffineExpr> unifiedResults = unifiedMap.getResults(); in compressUnusedImpl()
604 SmallVector<AffineExpr> symReplacements; in compressSymbols()
613 SmallVector<AffineExpr> resultExprs; in compressSymbols()
622 map.walkExprs([&](AffineExpr expr) { in compressUnusedSymbols()
635 SmallVector<AffineExpr, 8> exprs; in simplifyAffineMap()
646 SmallVector<AffineExpr, 4> uniqueExprs(results.begin(), results.end()); in removeDuplicateExprs()
657 SmallVector<AffineExpr, 4> exprs(map.getNumDims()); in inversePermutation()
667 SmallVector<AffineExpr, 4> seenExprs; in inversePermutation()
680 AffineExpr zero = mlir::getAffineConstantExpr(0, context); in inverseAndBroadcastProjectedPermutation()
682 SmallVector<AffineExpr, 4> exprs(map.getNumInputs(), zero); in inverseAndBroadcastProjectedPermutation()
702 SmallVector<AffineExpr, 8> results; in concatAffineMaps()