23 AffineExprKind AffineExpr::getKind() const { return expr->kind; }  in getKind()
43 AffineExpr mlir::getAffineBinaryOpExpr(AffineExprKind kind, AffineExpr lhs,  in getAffineBinaryOpExpr()
45   if (kind == AffineExprKind::Add)  in getAffineBinaryOpExpr()
47   if (kind == AffineExprKind::Mul)  in getAffineBinaryOpExpr()
49   if (kind == AffineExprKind::FloorDiv)  in getAffineBinaryOpExpr()
51   if (kind == AffineExprKind::CeilDiv)  in getAffineBinaryOpExpr()
53   if (kind == AffineExprKind::Mod)  in getAffineBinaryOpExpr()
65   case AffineExprKind::Constant:  in replaceDimsAndSymbols()
67   case AffineExprKind::DimId: {  in replaceDimsAndSymbols()
73   case AffineExprKind::SymbolId: {  in replaceDimsAndSymbols()
79   case AffineExprKind::Add:  in replaceDimsAndSymbols()
80   case AffineExprKind::Mul:  in replaceDimsAndSymbols()
81   case AffineExprKind::FloorDiv:  in replaceDimsAndSymbols()
82   case AffineExprKind::CeilDiv:  in replaceDimsAndSymbols()
83   case AffineExprKind::Mod:  in replaceDimsAndSymbols()
112   case AffineExprKind::Add:  in replace()
113   case AffineExprKind::Mul:  in replace()
114   case AffineExprKind::FloorDiv:  in replace()
115   case AffineExprKind::CeilDiv:  in replace()
116   case AffineExprKind::Mod:  in replace()
138   case AffineExprKind::Constant:  in isSymbolicOrConstant()
140   case AffineExprKind::DimId:  in isSymbolicOrConstant()
142   case AffineExprKind::SymbolId:  in isSymbolicOrConstant()
145   case AffineExprKind::Add:  in isSymbolicOrConstant()
146   case AffineExprKind::Mul:  in isSymbolicOrConstant()
147   case AffineExprKind::FloorDiv:  in isSymbolicOrConstant()
148   case AffineExprKind::CeilDiv:  in isSymbolicOrConstant()
149   case AffineExprKind::Mod: {  in isSymbolicOrConstant()
162   case AffineExprKind::SymbolId:  in isPureAffine()
163   case AffineExprKind::DimId:  in isPureAffine()
164   case AffineExprKind::Constant:  in isPureAffine()
166   case AffineExprKind::Add: {  in isPureAffine()
171   case AffineExprKind::Mul: {  in isPureAffine()
179   case AffineExprKind::FloorDiv:  in isPureAffine()
180   case AffineExprKind::CeilDiv:  in isPureAffine()
181   case AffineExprKind::Mod: {  in isPureAffine()
194   case AffineExprKind::SymbolId:  in getLargestKnownDivisor()
196   case AffineExprKind::DimId:  in getLargestKnownDivisor()
198   case AffineExprKind::Constant:  in getLargestKnownDivisor()
200   case AffineExprKind::Mul: {  in getLargestKnownDivisor()
205   case AffineExprKind::Add:  in getLargestKnownDivisor()
207   case AffineExprKind::FloorDiv:  in getLargestKnownDivisor()
208   case AffineExprKind::CeilDiv:  in getLargestKnownDivisor()
209   case AffineExprKind::Mod: {  in getLargestKnownDivisor()
223   case AffineExprKind::SymbolId:  in isMultipleOf()
225   case AffineExprKind::DimId:  in isMultipleOf()
227   case AffineExprKind::Constant:  in isMultipleOf()
229   case AffineExprKind::Mul: {  in isMultipleOf()
238   case AffineExprKind::Add:  in isMultipleOf()
239   case AffineExprKind::FloorDiv:  in isMultipleOf()
240   case AffineExprKind::CeilDiv:  in isMultipleOf()
241   case AffineExprKind::Mod: {  in isMultipleOf()
254   if (getKind() == AffineExprKind::DimId) {  in isFunctionOfDim()
286                                 AffineExprKind opKind) {  in isDivisibleBySymbol()
288   assert((opKind == AffineExprKind::Mod || opKind == AffineExprKind::FloorDiv ||  in isDivisibleBySymbol()
289           opKind == AffineExprKind::CeilDiv) &&  in isDivisibleBySymbol()
292   case AffineExprKind::Constant:  in isDivisibleBySymbol()
296   case AffineExprKind::DimId:  in isDivisibleBySymbol()
298   case AffineExprKind::SymbolId:  in isDivisibleBySymbol()
301   case AffineExprKind::Add: {  in isDivisibleBySymbol()
311   case AffineExprKind::Mod: {  in isDivisibleBySymbol()
314                                AffineExprKind::Mod) &&  in isDivisibleBySymbol()
316                                AffineExprKind::Mod);  in isDivisibleBySymbol()
319   case AffineExprKind::Mul: {  in isDivisibleBySymbol()
332   case AffineExprKind::FloorDiv:  in isDivisibleBySymbol()
333   case AffineExprKind::CeilDiv: {  in isDivisibleBySymbol()
347                                  AffineExprKind opKind) {  in symbolicDivide()
349   assert((opKind == AffineExprKind::Mod || opKind == AffineExprKind::FloorDiv ||  in symbolicDivide()
350           opKind == AffineExprKind::CeilDiv) &&  in symbolicDivide()
353   case AffineExprKind::Constant:  in symbolicDivide()
357   case AffineExprKind::DimId:  in symbolicDivide()
359   case AffineExprKind::SymbolId:  in symbolicDivide()
362   case AffineExprKind::Add: {  in symbolicDivide()
369   case AffineExprKind::Mod: {  in symbolicDivide()
377   case AffineExprKind::Mul: {  in symbolicDivide()
386   case AffineExprKind::FloorDiv:  in symbolicDivide()
387   case AffineExprKind::CeilDiv: {  in symbolicDivide()
405   case AffineExprKind::Constant:  in simplifySemiAffine()
406   case AffineExprKind::DimId:  in simplifySemiAffine()
407   case AffineExprKind::SymbolId:  in simplifySemiAffine()
409   case AffineExprKind::Add:  in simplifySemiAffine()
410   case AffineExprKind::Mul: {  in simplifySemiAffine()
421   case AffineExprKind::FloorDiv:  in simplifySemiAffine()
422   case AffineExprKind::CeilDiv:  in simplifySemiAffine()
423   case AffineExprKind::Mod: {  in simplifySemiAffine()
434     if (expr.getKind() == AffineExprKind::Mod)  in simplifySemiAffine()
442 static AffineExpr getAffineDimOrSymbol(AffineExprKind kind, unsigned position,  in getAffineDimOrSymbol()
454   return getAffineDimOrSymbol(AffineExprKind::DimId, position, context);  in getAffineDimExpr()
464   return getAffineDimOrSymbol(AffineExprKind::SymbolId, position, context);  in getAffineSymbolExpr()
512   if (lBin && rhsConst && lBin.getKind() == AffineExprKind::Add) {  in simplifyAdd()
524   if (lBinOpExpr && lBinOpExpr.getKind() == AffineExprKind::Mul &&  in simplifyAdd()
535   if (rBinOpExpr && rBinOpExpr.getKind() == AffineExprKind::Mul &&  in simplifyAdd()
546         AffineExprKind::Mul, firstExpr,  in simplifyAdd()
552   if (lBin && lBin.getKind() == AffineExprKind::Add) {  in simplifyAdd()
571   if (!lrBinOpExpr || lrBinOpExpr.getKind() != AffineExprKind::FloorDiv)  in simplifyAdd()
592       /*initFn=*/{}, static_cast<unsigned>(AffineExprKind::Add), *this, other);  in operator +()
627   if (lBin && rhsConst && lBin.getKind() == AffineExprKind::Mul) {  in simplifyMul()
634   if (lBin && lBin.getKind() == AffineExprKind::Mul) {  in simplifyMul()
652       /*initFn=*/{}, static_cast<unsigned>(AffineExprKind::Mul), *this, other);  in operator *()
686   if (lBin && lBin.getKind() == AffineExprKind::Mul) {  in simplifyFloorDiv()
696   if (lBin && lBin.getKind() == AffineExprKind::Add) {  in simplifyFloorDiv()
718       /*initFn=*/{}, static_cast<unsigned>(AffineExprKind::FloorDiv), *this,  in floorDiv()
741   if (lBin && lBin.getKind() == AffineExprKind::Mul) {  in simplifyCeilDiv()
761       /*initFn=*/{}, static_cast<unsigned>(AffineExprKind::CeilDiv), *this,  in ceilDiv()
786   if (lBin && lBin.getKind() == AffineExprKind::Add) {  in simplifyMod()
808       /*initFn=*/{}, static_cast<unsigned>(AffineExprKind::Mod), *this, other);  in operator %()