Lines Matching refs:SmallVector
162 SmallVector<unsigned> broadcastDims; in isPermutationOfMinorIdentityWithBroadcasting()
209 SmallVector<AffineExpr, 4> affExprs; in getPermutationMap()
219 static SmallVector<AffineMap, 4>
226 SmallVector<AffineMap, 4> maps; in inferFromExprList()
234 SmallVector<AffineMap, 4>
239 SmallVector<AffineMap, 4>
240 AffineMap::inferFromExprList(ArrayRef<SmallVector<AffineExpr, 4>> exprsList) { in inferFromExprList()
246 SmallVector<AffineExpr, 4> dimExprs; in getMultiDimIdentityMap()
286 SmallVector<int64_t> AffineMap::getConstantResults() const { in getConstantResults()
288 SmallVector<int64_t> result; in getConstantResults()
334 SmallVector<int64_t, 2> integers; in constantFold()
356 SmallVector<AffineExpr, 4> exprs; in partialConstantFold()
396 SmallVector<AffineExpr, 8> results; in replaceDimsAndSymbols()
410 SmallVector<AffineExpr, 4> newResults; in replace()
423 SmallVector<AffineExpr, 4> newResults; in replace()
432 SmallVector<AffineExpr, 4> newResults; 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()
463 SmallVector<int64_t, 4> AffineMap::compose(ArrayRef<int64_t> values) const { in compose()
465 SmallVector<AffineExpr, 4> exprs; in compose()
471 SmallVector<int64_t, 4> res; in compose()
487 SmallVector<bool, 8> seen(getNumInputs(), false); in isProjectedPermutation()
515 SmallVector<AffineExpr, 4> exprs; in getSubMap()
546 SmallVector<AffineExpr> dimReplacements; in compressDims()
555 SmallVector<AffineExpr> resultExprs; in compressDims()
566 static SmallVector<AffineMap>
570 return SmallVector<AffineMap>(); in compressUnusedImpl()
571 SmallVector<AffineExpr> allExprs; in compressUnusedImpl()
585 SmallVector<AffineMap> res; in compressUnusedImpl()
596 SmallVector<AffineMap> mlir::compressUnusedDims(ArrayRef<AffineMap> maps) { in compressUnusedDims()
604 SmallVector<AffineExpr> symReplacements; in compressSymbols()
613 SmallVector<AffineExpr> resultExprs; in compressSymbols()
629 SmallVector<AffineMap> mlir::compressUnusedSymbols(ArrayRef<AffineMap> maps) { 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()
682 SmallVector<AffineExpr, 4> exprs(map.getNumInputs(), zero); in inverseAndBroadcastProjectedPermutation()
702 SmallVector<AffineExpr, 8> results; in concatAffineMaps()