tvm
Loading...
Searching...
No Matches
Public Member Functions | Static Public Member Functions | Public Attributes | Static Public Attributes | List of all members
tvm::s_tir::IterVarNode Class Reference

An iteration variable representing an iteration over a one dimensional interval. More...

#include <iter_var.h>

Inheritance diagram for tvm::s_tir::IterVarNode:
Collaboration diagram for tvm::s_tir::IterVarNode:

Public Member Functions

 IterVarNode (ffi::UnsafeInit tag)
 
 IterVarNode (PrimVar var)
 
 TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("s_tir.IterVar", IterVarNode, OpaqueExprNode)
 
- Public Member Functions inherited from tvm::OpaqueExprNode
 TVM_FFI_DECLARE_OBJECT_INFO ("ir.OpaqueExpr", OpaqueExprNode, ExprNode)
 
- Public Member Functions inherited from tvm::ExprNode
 TVM_FFI_DECLARE_OBJECT_INFO ("ir.Expr", ExprNode, ffi::Object)
 

Static Public Member Functions

static void RegisterReflection ()
 
- Static Public Member Functions inherited from tvm::OpaqueExprNode
static void RegisterReflection ()
 
- Static Public Member Functions inherited from tvm::ExprNode
static void RegisterReflection ()
 

Public Attributes

ffi::Optional< Range > dom
 the domain of iteration, if known, can be None For the intermediate schedule node, before schedule.
 
PrimVar var
 The looping variable.
 
IterVarType iter_type
 The type of the IterVar.
 
ffi::String thread_tag
 additional tag on the iteration variable, set this if this is bound already to a known thread tag.
 
- Public Attributes inherited from tvm::ExprNode
Location loc = UnknownLoc()
 Location that points to the original source code. Reserved debug information.
 
Type ty = MissingType()
 The deduced or annotated type of the expression.
 

Static Public Attributes

static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode
 
- Static Public Attributes inherited from tvm::OpaqueExprNode
static constexpr const uint32_t _type_child_slots = 2
 
- Static Public Attributes inherited from tvm::ExprNode
static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode
 
static constexpr const uint32_t _type_child_slots = 64
 

Detailed Description

An iteration variable representing an iteration over a one dimensional interval.

The dtype of the extent of the dom of the IterVar must match the dtype of the internal Var.

Constructor & Destructor Documentation

◆ IterVarNode() [1/2]

tvm::s_tir::IterVarNode::IterVarNode ( ffi::UnsafeInit  tag)
inlineexplicit

◆ IterVarNode() [2/2]

tvm::s_tir::IterVarNode::IterVarNode ( PrimVar  var)
inlineexplicit

Member Function Documentation

◆ RegisterReflection()

static void tvm::s_tir::IterVarNode::RegisterReflection ( )
inlinestatic

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::s_tir::IterVarNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL ( "s_tir.IterVar"  ,
IterVarNode  ,
OpaqueExprNode   
)

Member Data Documentation

◆ _type_s_eq_hash_kind

constexpr TVMFFISEqHashKind tvm::s_tir::IterVarNode::_type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode
staticconstexpr

◆ dom

ffi::Optional<Range> tvm::s_tir::IterVarNode::dom

the domain of iteration, if known, can be None For the intermediate schedule node, before schedule.

◆ iter_type

IterVarType tvm::s_tir::IterVarNode::iter_type

The type of the IterVar.

◆ thread_tag

ffi::String tvm::s_tir::IterVarNode::thread_tag

additional tag on the iteration variable, set this if this is bound already to a known thread tag.

◆ var

PrimVar tvm::s_tir::IterVarNode::var

The looping variable.


The documentation for this class was generated from the following file: