20   return SymVar(llvm::cast<AffineSymbolExpr>(expr));  in castSymVar()
24   if (const auto s = dyn_cast_or_null<AffineSymbolExpr>(expr))  in dyn_castSymVar()
30   return Var(getAllowedVarKind(), llvm::cast<AffineDimExpr>(expr));  in castDimLvlVar()
34   if (const auto x = dyn_cast_or_null<AffineDimExpr>(expr))  in dyn_castDimLvlVar()
42   const auto binop = llvm::dyn_cast<AffineBinaryOpExpr>(expr);  in unpackBinop()
52 DimSpec::DimSpec(DimVar var, DimExpr expr, SparseTensorDimSliceAttr slice)  in DimSpec()  argument
53     : var(var), expr(expr), slice(slice) {}  in DimSpec()
58   return ranks.isValid(var) && (!expr || ranks.isValid(expr));  in isValid()
65 LvlSpec::LvlSpec(LvlVar var, LvlExpr expr, LevelType type)  in LvlSpec()  argument
66     : var(var), expr(expr), type(type) {  in LvlSpec()
67   assert(expr);  in LvlSpec()
73   return ranks.isValid(var) && ranks.isValid(expr);  in isValid()
131     auto expr = dimSpec.getExpr().getAffineExpr();  in getLvlToDimMap()  local
132     if (expr) {  in getLvlToDimMap()
133       dimAffines.push_back(expr);  in getLvlToDimMap()