| /llvm-project-15.0.7/mlir/test/python/ir/ |
| H A D | affine_map.py | 184 s0, s1, s2 = (AffineSymbolExpr.get(0), AffineSymbolExpr.get(1),
185 AffineSymbolExpr.get(2))
211 s0, s1, s2 = (AffineSymbolExpr.get(0), AffineSymbolExpr.get(1),
212 AffineSymbolExpr.get(2))
|
| H A D | integer_set.py | 33 s0 = AffineSymbolExpr.get(0)
45 set2 = set0.get_replaced([d0, AffineSymbolExpr.get(1)], [s0], 1, 2)
103 s0 = AffineSymbolExpr.get(0)
|
| H A D | affine_expr.py | 109 s11 = AffineSymbolExpr.get(1)
110 s2 = AffineSymbolExpr.get(2)
378 map1 = AffineAddExpr.get(AffineDimExpr.get(0), AffineSymbolExpr.get(1))
379 map2 = AffineAddExpr.get(AffineDimExpr.get(1), AffineSymbolExpr.get(0))
394 s1 = AffineSymbolExpr.get(1)
|
| /llvm-project-15.0.7/mlir/include/mlir/IR/ |
| H A D | AffineExprVisitor.h | 112 expr.cast<AffineSymbolExpr>()); in walkPostOrder()
149 expr.cast<AffineSymbolExpr>()); in visit()
182 RetTy visitSymbolExpr(AffineSymbolExpr expr) { return RetTy(); } in visitSymbolExpr()
281 void visitSymbolExpr(AffineSymbolExpr expr);
|
| H A D | AffineExpr.h | 224 class AffineSymbolExpr : public AffineExpr {
227 /* implicit */ AffineSymbolExpr(AffineExpr::ImplType *ptr);
275 if (std::is_same<U, AffineSymbolExpr>::value) in isa()
|
| H A D | AffineMap.h | 577 if (auto s = e.dyn_cast<AffineSymbolExpr>()) in getMaxDimAndSymbol()
|
| /llvm-project-15.0.7/mlir/lib/IR/ |
| H A D | AffineExpr.cpp | 38 void visitSymbolExpr(AffineSymbolExpr expr) { callback(expr); } in walk()
76 unsigned symId = cast<AffineSymbolExpr>().getPosition(); in replaceDimsAndSymbols()
335 return (expr.cast<AffineSymbolExpr>().getPosition() == symbolPos); in isDivisibleBySymbol()
463 AffineSymbolExpr symbolExpr = in simplifySemiAffine()
493 AffineSymbolExpr::AffineSymbolExpr(AffineExpr::ImplType *ptr) in AffineSymbolExpr() function in AffineSymbolExpr
495 unsigned AffineSymbolExpr::getPosition() const { in getPosition()
1029 (rhs.isa<AffineDimExpr>() || rhs.isa<AffineSymbolExpr>() || in getSemiAffineExprFromFlatForm()
1044 lhsPos = lhs.cast<AffineSymbolExpr>().getPosition(); in getSemiAffineExprFromFlatForm()
1057 rhsPos = rhs.cast<AffineSymbolExpr>().getPosition(); in getSemiAffineExprFromFlatForm()
1065 lhsPos = lhs.cast<AffineSymbolExpr>().getPosition(); in getSemiAffineExprFromFlatForm()
[all …]
|
| H A D | AffineMap.cpp | 68 expr.cast<AffineSymbolExpr>().getPosition()] in constantFoldImpl()
623 if (auto symExpr = expr.dyn_cast<AffineSymbolExpr>()) in compressUnusedSymbols()
|
| H A D | AsmPrinter.cpp | 2352 unsigned pos = expr.cast<AffineSymbolExpr>().getPosition(); in printAffineExprInternal()
|
| /llvm-project-15.0.7/mlir/lib/CAPI/IR/ |
| H A D | AffineExpr.cpp | 85 return unwrap(affineExpr).isa<AffineSymbolExpr>(); in mlirAffineExprIsASymbol()
93 return unwrap(affineExpr).cast<AffineSymbolExpr>().getPosition(); in mlirAffineSymbolExprGetPosition()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgInterfaces.cpp | 210 LogicalResult visitSymbolExpr(AffineSymbolExpr expr) { return failure(); } in visitSymbolExpr()
237 getAffineExprOfType<AffineSymbolExpr>(lhsExpr, rhsExpr); in isDimExprOrMulExpr()
556 bool visitSymbolExpr(AffineSymbolExpr symbolExpr) { return false; } in visitSymbolExpr()
|
| /llvm-project-15.0.7/mlir/python/mlir/dialects/linalg/opdsl/lang/ |
| H A D | affine.py | 290 return _ir.AffineSymbolExpr.get(position=pos)
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Utils/ |
| H A D | ReshapeOpsUtils.cpp | 185 assert(getMaxPosOfType<AffineSymbolExpr>(reassociation) == 0 && in getSymbolLessAffineMaps()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Analysis/ |
| H A D | LoopAnalysis.cpp | 235 else if (auto symExpr = expr.dyn_cast<AffineSymbolExpr>()) in isContiguousAccess()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Loops.cpp | 292 if (expr.dyn_cast<AffineDimExpr>() || expr.dyn_cast<AffineSymbolExpr>()) { in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/python/mlir/_mlir_libs/_mlir/ |
| H A D | ir.pyi | 30 "AffineSymbolExpr",
350 class AffineSymbolExpr(AffineExpr):
353 def get(*args, **kwargs) -> AffineSymbolExpr: ...
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/IR/ |
| H A D | AffineOps.cpp | 575 if (auto sym = expr.dyn_cast<AffineSymbolExpr>()) in fold()
998 else if (auto symExpr = expr.dyn_cast<AffineSymbolExpr>()) in canonicalizeMapOrSetAndOperands()
1715 if (auto symExpr = expr.dyn_cast<AffineSymbolExpr>()) { in printBound()
2918 if (auto symExpr = expr.dyn_cast<AffineSymbolExpr>()) { in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/tools/mlir-linalg-ods-gen/ |
| H A D | mlir-linalg-ods-yaml-gen.cpp | 868 if (auto symbol = en.value().dyn_cast<AffineSymbolExpr>()) { in generateNamedGenericOpDefns()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/ |
| H A D | Utils.cpp | 198 Value visitSymbolExpr(AffineSymbolExpr expr) { in visitSymbolExpr()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/IR/ |
| H A D | LinalgInterfaces.td | 100 MulExpr ::== AffineDimExpr (`*` (AffineConstantExpr | AffineSymbolExpr))?
|