• Home
  • Raw
  • Download

Lines Matching refs:AffineMap

94 AffineMap AffineMap::getConstantMap(int64_t val, MLIRContext *context) {  in getConstantMap()
101 AffineMap AffineMap::getMinorIdentityMap(unsigned dims, unsigned results, in getMinorIdentityMap()
104 auto id = AffineMap::getMultiDimIdentityMap(dims, context); in getMinorIdentityMap()
105 return AffineMap::get(dims, 0, id.getResults().take_back(results), context); in getMinorIdentityMap()
108 bool AffineMap::isMinorIdentity() const { in isMinorIdentity()
114 AffineMap AffineMap::getPermutationMap(ArrayRef<unsigned> permutation, in getPermutationMap()
122 auto permutationMap = AffineMap::get(*m + 1, 0, affExprs, context); in getPermutationMap()
143 static SmallVector<AffineMap, 4>
150 SmallVector<AffineMap, 4> maps; in inferFromExprList()
153 maps.push_back(AffineMap::get(/*dimCount=*/maxDim + 1, in inferFromExprList()
158 SmallVector<AffineMap, 4>
159 AffineMap::inferFromExprList(ArrayRef<ArrayRef<AffineExpr>> exprsList) { in inferFromExprList()
163 SmallVector<AffineMap, 4>
164 AffineMap::inferFromExprList(ArrayRef<SmallVector<AffineExpr, 4>> exprsList) { in inferFromExprList()
168 AffineMap AffineMap::getMultiDimIdentityMap(unsigned numDims, in getMultiDimIdentityMap()
177 MLIRContext *AffineMap::getContext() const { return map->context; } in getContext()
179 bool AffineMap::isIdentity() const { in isIdentity()
191 bool AffineMap::isEmpty() const { in isEmpty()
195 bool AffineMap::isSingleConstant() const { in isSingleConstant()
199 int64_t AffineMap::getSingleConstantResult() const { in getSingleConstantResult()
204 unsigned AffineMap::getNumDims() const { in getNumDims()
208 unsigned AffineMap::getNumSymbols() const { in getNumSymbols()
212 unsigned AffineMap::getNumResults() const { in getNumResults()
216 unsigned AffineMap::getNumInputs() const { in getNumInputs()
221 ArrayRef<AffineExpr> AffineMap::getResults() const { in getResults()
225 AffineExpr AffineMap::getResult(unsigned idx) const { in getResult()
230 unsigned AffineMap::getDimPosition(unsigned idx) const { in getDimPosition()
238 AffineMap::constantFold(ArrayRef<Attribute> operandConstants, in constantFold()
256 AffineMap
257 AffineMap::partialConstantFold(ArrayRef<Attribute> operandConstants, in partialConstantFold()
289 void AffineMap::walkExprs(std::function<void(AffineExpr)> callback) const { in walkExprs()
299 AffineMap AffineMap::replaceDimsAndSymbols(ArrayRef<AffineExpr> dimReplacements, in replaceDimsAndSymbols()
312 AffineMap AffineMap::compose(AffineMap map) { in compose()
333 return AffineMap::get(numDims, numSymbols, exprs, map.getContext()); in compose()
336 SmallVector<int64_t, 4> AffineMap::compose(ArrayRef<int64_t> values) { in compose()
343 auto resMap = compose(AffineMap::get(0, 0, exprs, ctx)); in compose()
351 bool AffineMap::isProjectedPermutation() { in isProjectedPermutation()
367 bool AffineMap::isPermutation() { in isPermutation()
373 AffineMap AffineMap::getSubMap(ArrayRef<unsigned> resultPos) { in getSubMap()
378 return AffineMap::get(getNumDims(), getNumSymbols(), exprs, getContext()); in getSubMap()
381 AffineMap AffineMap::getMajorSubMap(unsigned numResults) { in getMajorSubMap()
383 return AffineMap(); in getMajorSubMap()
389 AffineMap AffineMap::getMinorSubMap(unsigned numResults) { in getMinorSubMap()
391 return AffineMap(); in getMinorSubMap()
398 AffineMap mlir::simplifyAffineMap(AffineMap map) { in simplifyAffineMap()
404 return AffineMap::get(map.getNumDims(), map.getNumSymbols(), exprs, in simplifyAffineMap()
408 AffineMap mlir::removeDuplicateExprs(AffineMap map) { in removeDuplicateExprs()
413 return AffineMap::get(map.getNumDims(), map.getNumSymbols(), uniqueExprs, in removeDuplicateExprs()
417 AffineMap mlir::inversePermutation(AffineMap map) { in inversePermutation()
437 return AffineMap(); in inversePermutation()
438 return AffineMap::get(map.getNumResults(), 0, seenExprs, map.getContext()); in inversePermutation()
441 AffineMap mlir::concatAffineMaps(ArrayRef<AffineMap> maps) { in concatAffineMaps()
454 return AffineMap::get(numDims, numSymbols, results, in concatAffineMaps()
458 AffineMap mlir::getProjectedMap(AffineMap map, in getProjectedMap()
470 return map.compose(AffineMap::get( in getProjectedMap()
478 MutableAffineMap::MutableAffineMap(AffineMap map) in MutableAffineMap()
485 void MutableAffineMap::reset(AffineMap map) { in reset()
513 AffineMap MutableAffineMap::getAffineMap() const { in getAffineMap()
514 return AffineMap::get(numDims, numSymbols, results, context); in getAffineMap()