Module expr

Expressions: variables, constants, calls, tuples and tensor accesses.

The objects python/tvm_tirx/ir/expr.py defines. Expr is the base of every expression and carries its type and location. The primitive arithmetic and vector expressions are in prim. The constructors are new, or from_dtype and the like where new allocates the fields as given; they are written in Rust after the C++ constructors, with their checks, defaults and errors. The constants of the common types are IntImm::int32, IntImm::int64, IntImm::bool, FloatImm::float32 and FloatImm::float64, and an i32, i64, bool, f32 or f64 converts into an Expr as C++ converts it implicitly.

Structs

• Call
• CallObj
  Complete: reflected fields fill [40, 72) exactly.
• Constant
• ConstantObj
  Complete: reflected fields fill [40, 40) exactly.
• DataTypeImm
• DataTypeImmObj
  Complete: reflected fields fill [40, 48) exactly (alignment padding [44, 48)).
• Expr
• ExprObj
  Complete: reflected fields fill [24, 40) exactly.
• FloatImm
• FloatImmObj
  Complete: reflected fields fill [40, 48) exactly.
• GenericConst
• GenericConstObj
  Complete: reflected fields fill [40, 56) exactly.
• GlobalVar
• GlobalVarObj
  Complete: reflected fields fill [40, 56) exactly.
• IntExpr
  A PrimExpr of a scalar signed or unsigned integer type of any width (C++ IntExpr), such as an index or an extent.
• IntImm
• IntImmObj
  Complete: reflected fields fill [40, 56) exactly.
• LambdaExpr
• LambdaExprObj
  Complete: reflected fields fill [40, 56) exactly.
• OpaqueExpr
• OpaqueExprObj
  Complete: reflected fields fill [40, 40) exactly.
• PrimExpr
  An expression typed by a PrimType (C++ PrimExpr): a constant, a variable, an operator on scalars or vectors, or a call that returns one.
• Range
• RangeObj
  Complete: reflected fields fill [24, 48) exactly.
• StagingExpr
• StagingExprObj
  Complete: reflected fields fill [40, 40) exactly.
• StringImm
• StringImmObj
  Complete: reflected fields fill [40, 56) exactly.
• TensorLoad
• TensorLoadObj
  Complete: reflected fields fill [40, 56) exactly.
• TensorRegion
• TensorRegionObj
  Complete: reflected fields fill [40, 56) exactly.
• Tuple
• TupleGetItem
• TupleGetItemObj
  Complete: reflected fields fill [40, 56) exactly (alignment padding [52, 56)).
• TupleObj
  Complete: reflected fields fill [40, 48) exactly.
• Var
• VarObj
  Complete: reflected fields fill [40, 56) exactly.

Enums

• Subscript
  One index of Expr::subscript: an expression, or a start:stop:step slice whose missing bounds are None.

Traits

• IntoExpr
  What converts to an expression, possibly failing: anything that converts into an Expr (a node, a Var, a Rust i32, f32 and the like), and an ExprResult, an expression under construction, which gives the first error any of its steps raised. The constructors and functions that can fail take their expression operands as impl IntoExpr, so an ExprResult passes straight in and ? on their result covers it.