Lines Matching refs:AffineMap
92 AffineMap map = op.getAffineMap(); in remainsLegalAfterInline()
524 AffineMap affineMap = getMap(); in verify()
595 static LogicalResult replaceDimOrSym(AffineMap *map, in replaceDimOrSym()
615 AffineMap composeMap = affineApply.getAffineMap(); in replaceDimOrSym()
638 static void composeAffineMapAndOperands(AffineMap *map, in composeAffineMapAndOperands()
700 void mlir::fullyComposeAffineMapAndOperands(AffineMap *map, in fullyComposeAffineMapAndOperands()
777 AffineMap map, in makeComposedAffineApply()
779 AffineMap normalizedMap = map; in makeComposedAffineApply()
789 b, loc, AffineMap::inferFromExprList(ArrayRef<AffineExpr>{e}).front(), in makeComposedAffineApply()
795 AffineMap map, in makeComposedFoldedAffineApply()
816 b, loc, AffineMap::inferFromExprList(ArrayRef<AffineExpr>{expr}).front(), in makeComposedFoldedAffineApply()
822 static void composeMultiResultAffineMap(AffineMap &map, in composeMultiResultAffineMap()
831 AffineMap submap = map.getSubMap({i}); in composeMultiResultAffineMap()
846 map = AffineMap::get(dims.size(), symbols.size(), exprs, map.getContext()); in composeMultiResultAffineMap()
850 Value mlir::makeComposedAffineMin(OpBuilder &b, Location loc, AffineMap map, in makeComposedAffineMin()
858 mlir::makeComposedFoldedAffineMin(RewriterBase &b, Location loc, AffineMap map, in makeComposedFoldedAffineMin()
876 AffineMap map, in createFoldedComposedAffineApply()
885 AffineMap map, ValueRange values) { in applyMapToValues()
893 AffineMap map = AffineMap::get(numDims, numSym, expr); in applyMapToValues()
900 mlir::applyMapToValues(RewriterBase &b, Location loc, AffineMap map, in applyMapToValues()
914 AffineMap submap = map.getSubMap({i}); in applyMapToValues()
981 static_assert(llvm::is_one_of<MapOrSet, AffineMap, IntegerSet>::value, in canonicalizeMapOrSetAndOperands()
1055 void mlir::canonicalizeMapAndOperands(AffineMap *map, in canonicalizeMapAndOperands()
1057 canonicalizeMapOrSetAndOperands<AffineMap>(map, operands); in canonicalizeMapAndOperands()
1076 AffineMap map, ArrayRef<Value> mapOperands) const;
1087 AffineMap oldMap = map; in matchAndRewrite()
1105 PatternRewriter &rewriter, AffineLoadOp load, AffineMap map, in replaceAffineOp()
1112 PatternRewriter &rewriter, AffinePrefetchOp prefetch, AffineMap map, in replaceAffineOp()
1121 PatternRewriter &rewriter, AffineStoreOp store, AffineMap map, in replaceAffineOp()
1128 PatternRewriter &rewriter, AffineVectorLoadOp vectorload, AffineMap map, in replaceAffineOp()
1136 PatternRewriter &rewriter, AffineVectorStoreOp vectorstore, AffineMap map, in replaceAffineOp()
1146 PatternRewriter &rewriter, AffineOpTy op, AffineMap map, in replaceAffineOp()
1183 Value srcMemRef, AffineMap srcMap, in build()
1185 AffineMap dstMap, ValueRange destIndices, in build()
1186 Value tagMemRef, AffineMap tagMap, in build()
1352 Value tagMemRef, AffineMap tagMap, in build()
1431 ValueRange lbOperands, AffineMap lbMap, in build()
1432 ValueRange ubOperands, AffineMap ubMap, int64_t step, in build()
1484 auto lbMap = AffineMap::getConstantMap(lb, builder.getContext()); in build()
1485 auto ubMap = AffineMap::getConstantMap(ub, builder.getContext()); in build()
1559 AffineMap map = builder.getSymbolIdentityMap(); in parseBound()
1693 AffineMap map = boundMap.getValue(); in printBound()
1732 AffineMap lbMap = getLowerBoundMapAttr().getValue(); in getNumIterOperands()
1733 AffineMap ubMap = getUpperBoundMapAttr().getValue(); in getNumIterOperands()
1786 AffineMap boundMap = in foldLoopBounds()
1999 void AffineForOp::setLowerBound(ValueRange lbOperands, AffineMap map) { in setLowerBound()
2014 void AffineForOp::setUpperBound(ValueRange ubOperands, AffineMap map) { in setUpperBound()
2027 void AffineForOp::setLowerBoundMap(AffineMap map) { in setLowerBoundMap()
2036 void AffineForOp::setUpperBoundMap(AffineMap map) { in setUpperBoundMap()
2062 setLowerBound({}, AffineMap::getConstantMap(value, getContext())); in setConstantLowerBound()
2066 setUpperBound({}, AffineMap::getConstantMap(value, getContext())); in setConstantUpperBound()
2503 auto map = AffineMap::get(set.getNumDims(), set.getNumSymbols(), in composeSetAndOperands()
2541 AffineMap map, ValueRange operands) { in build()
2551 Value memref, AffineMap map, ValueRange mapOperands) { in build()
2609 AffineMap map = mapAttr.getValue(); in verifyMemoryOpIndexing()
2691 Value valueToStore, Value memref, AffineMap map, in build()
2860 AffineMap oldMap = affineOp.getAffineMap(); in matchAndRewrite()
2873 auto newMap = AffineMap::get(oldMap.getNumDims(), oldMap.getNumSymbols(), in matchAndRewrite()
2903 AffineMap oldMap = affineOp.getAffineMap(); in matchAndRewrite()
2945 AffineMap producerMap = producerOp.getAffineMap(); in matchAndRewrite()
2967 auto newMap = AffineMap::get(numUsedDims, numUsedSyms, newExprs, in matchAndRewrite()
2985 static LogicalResult canonicalizeMapExprAndTermOrder(AffineMap &map) { in canonicalizeMapExprAndTermOrder()
3018 map = AffineMap::get(map.getNumDims(), map.getNumSymbols(), newExprs, in canonicalizeMapExprAndTermOrder()
3042 AffineMap map = affineOp.getAffineMap(); in matchAndRewrite()
3196 AffineMap map = mapAttr.getValue(); in verify()
3235 SmallVector<AffineMap> lbs(ranges.size(), builder.getConstantAffineMap(0)); in build()
3247 ArrayRef<AffineMap> lbMaps, ValueRange lbArgs, in build()
3248 ArrayRef<AffineMap> ubMaps, ValueRange ubArgs, in build()
3251 [lbMaps](AffineMap m) { in build()
3258 [ubMaps](AffineMap m) { in build()
3283 auto concatMapsSameInput = [&builder](ArrayRef<AffineMap> maps, in build()
3286 return AffineMap::get(builder.getContext()); in build()
3290 for (AffineMap m : maps) { in build()
3294 return AffineMap::get(maps[0].getNumDims(), maps[0].getNumSymbols(), exprs, in build()
3300 AffineMap lbMap = concatMapsSameInput(lbMaps, lbGroups); in build()
3301 AffineMap ubMap = concatMapsSameInput(ubMaps, ubGroups); in build()
3337 AffineMap AffineParallelOp::getLowerBoundMap(unsigned pos) { in getLowerBoundMap()
3345 AffineMap AffineParallelOp::getUpperBoundMap(unsigned pos) { in getUpperBoundMap()
3387 void AffineParallelOp::setLowerBounds(ValueRange lbOperands, AffineMap map) { in setLowerBounds()
3400 void AffineParallelOp::setUpperBounds(ValueRange ubOperands, AffineMap map) { in setUpperBounds()
3411 void AffineParallelOp::setLowerBoundsMap(AffineMap map) { in setLowerBoundsMap()
3412 AffineMap lbMap = getLowerBoundsMap(); in setLowerBoundsMap()
3419 void AffineParallelOp::setUpperBoundsMap(AffineMap map) { in setUpperBoundsMap()
3420 AffineMap ubMap = getUpperBoundsMap(); in setUpperBoundsMap()
3524 AffineMap map = mapAttr.getValue(); in printMinMaxBound()
3539 AffineMap submap = map.getSliceMap(start, size); in printMinMaxBound()
3722 auto flatMap = AffineMap::get(totalNumDims, totalNumSyms, flatExprs, in parseAffineMapWithMinMax()
3852 VectorType resultType, AffineMap map, in build()
3863 AffineMap map, ValueRange mapOperands) { in build()
3952 Value valueToStore, Value memref, AffineMap map, in build()