Class PrimExpr

• Defined in File expr.h

Inheritance Relationships

Base Type

• public tvm::TypedExpr< PrimType > (Template Class TypedExpr)

Derived Types

• public tvm::FloatImm (Class FloatImm)

• public tvm::IntExpr (Class IntExpr)

• public tvm::IntImm (Class IntImm)

• public tvm::PrimVar (Class PrimVar)

• public tvm::TensorLoad (Class TensorLoad)

• public tvm::prim::Add (Class Add)

• public tvm::prim::And (Class And)

• public tvm::prim::BitwiseAnd (Class BitwiseAnd)

• public tvm::prim::BitwiseNot (Class BitwiseNot)

• public tvm::prim::BitwiseOr (Class BitwiseOr)

• public tvm::prim::BitwiseXor (Class BitwiseXor)

• public tvm::prim::Broadcast (Class Broadcast)

• public tvm::prim::Cast (Class Cast)

• public tvm::prim::Div (Class Div)

• public tvm::prim::EQ (Class EQ)

• public tvm::prim::FloorDiv (Class FloorDiv)

• public tvm::prim::FloorMod (Class FloorMod)

• public tvm::prim::GE (Class GE)

• public tvm::prim::GT (Class GT)

• public tvm::prim::LE (Class LE)

• public tvm::prim::LShift (Class LShift)

• public tvm::prim::LT (Class LT)

• public tvm::prim::Let (Class Let)

• public tvm::prim::Max (Class Max)

• public tvm::prim::Min (Class Min)

• public tvm::prim::Mod (Class Mod)

• public tvm::prim::Mul (Class Mul)

• public tvm::prim::NE (Class NE)

• public tvm::prim::Not (Class Not)

• public tvm::prim::Or (Class Or)

• public tvm::prim::RShift (Class RShift)

• public tvm::prim::Ramp (Class Ramp)

• public tvm::prim::Select (Class Select)

• public tvm::prim::Shuffle (Class Shuffle)

• public tvm::prim::Sub (Class Sub)

• public tvm::sym::IterMapExpr (Class IterMapExpr)

Class Documentation

class PrimExpr : public tvm::TypedExpr<PrimType>

Typed reference/view over any Expr whose ExprObj::ty is PrimType.

PrimExpr is a type category rather than a dedicated runtime node category. It can contain intrinsic primitive nodes such as IntImmObj and FloatImmObj, or a general ExprObj such as CallObj, when that expression’s ty field is a PrimType. This keeps primitive-only APIs explicit while allowing shared Expr nodes for cross-dialect values with richer result types when needed.

Subclassed by tvm::FloatImm, tvm::IntExpr, tvm::IntImm, tvm::PrimVar, tvm::TensorLoad, tvm::prim::Add, tvm::prim::And, tvm::prim::BitwiseAnd, tvm::prim::BitwiseNot, tvm::prim::BitwiseOr, tvm::prim::BitwiseXor, tvm::prim::Broadcast, tvm::prim::Cast, tvm::prim::Div, tvm::prim::EQ, tvm::prim::FloorDiv, tvm::prim::FloorMod, tvm::prim::GE, tvm::prim::GT, tvm::prim::LE, tvm::prim::LShift, tvm::prim::LT, tvm::prim::Let, tvm::prim::Max, tvm::prim::Min, tvm::prim::Mod, tvm::prim::Mul, tvm::prim::NE, tvm::prim::Not, tvm::prim::Or, tvm::prim::RShift, tvm::prim::Ramp, tvm::prim::Select, tvm::prim::Shuffle, tvm::prim::Sub, tvm::sym::IterMapExpr

Public Types

using ContainerType = ExprObj

Public Functions

PrimExpr(int32_t value)

construct from integer.

Parameters:

value – The value to be constructed.

PrimExpr(float value)

construct from float.

Parameters:

value – The value to be constructed.

inline explicit PrimExpr(ffi::ObjectPtr<ExprObj> node)

inline explicit PrimExpr(::tvm::ffi::UnsafeInit tag)

inline __PtrType operator->() const

inline __PtrType get() const