tvm
Public Member Functions | List of all members
tvm::tir::IterVar Class Reference

Iteration Variable, represents an iteration over an integer interval. More...

#include <var.h>

Inheritance diagram for tvm::tir::IterVar:
Collaboration diagram for tvm::tir::IterVar:

Public Member Functions

 IterVar (Range dom, Var var, IterVarType iter_type, String thread_tag="", Span span=Span())
 
 operator PrimExpr () const
 
 TVM_DEFINE_OBJECT_REF_METHODS (IterVar, ObjectRef, IterVarNode)
 
 TVM_DEFINE_OBJECT_REF_COW_METHOD (IterVarNode)
 
- Public Member Functions inherited from tvm::runtime::ObjectRef
 ObjectRef ()=default
 default constructor More...
 
 ObjectRef (ObjectPtr< Object > data)
 Constructor from existing object ptr. More...
 
bool same_as (const ObjectRef &other) const
 Comparator. More...
 
bool operator== (const ObjectRef &other) const
 Comparator. More...
 
bool operator!= (const ObjectRef &other) const
 Comparator. More...
 
bool operator< (const ObjectRef &other) const
 Comparator. More...
 
bool defined () const
 
const Objectget () const
 
const Objectoperator-> () const
 
bool unique () const
 
int use_count () const
 
template<typename ObjectType , typename = std::enable_if_t<std::is_base_of_v<Object, ObjectType>>>
const ObjectType * as () const
 Try to downcast the internal Object to a raw pointer of a corresponding type. More...
 
template<typename ObjectRefType , typename = std::enable_if_t<std::is_base_of_v<ObjectRef, ObjectRefType>>>
Optional< ObjectRefType > as () const
 Try to downcast the ObjectRef to a Optional<T> of the requested type. More...
 

Additional Inherited Members

- Public Types inherited from tvm::runtime::ObjectRef
using ContainerType = Object
 type indicate the container type. More...
 
- Static Public Attributes inherited from tvm::runtime::ObjectRef
static constexpr bool _type_is_nullable = true
 
- Protected Member Functions inherited from tvm::runtime::ObjectRef
Objectget_mutable () const
 
- Static Protected Member Functions inherited from tvm::runtime::ObjectRef
template<typename T >
static T DowncastNoCheck (ObjectRef ref)
 Internal helper function downcast a ref without check. More...
 
static void FFIClearAfterMove (ObjectRef *ref)
 Clear the object ref data field without DecRef after we successfully moved the field. More...
 
template<typename ObjectType >
static ObjectPtr< ObjectType > GetDataPtr (const ObjectRef &ref)
 Internal helper function get data_ as ObjectPtr of ObjectType. More...
 
- Protected Attributes inherited from tvm::runtime::ObjectRef
ObjectPtr< Objectdata_
 Internal pointer that backs the reference. More...
 

Detailed Description

Iteration Variable, represents an iteration over an integer interval.

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

Constructor & Destructor Documentation

◆ IterVar()

tvm::tir::IterVar::IterVar ( Range  dom,
Var  var,
IterVarType  iter_type,
String  thread_tag = "",
Span  span = Span() 
)

Member Function Documentation

◆ operator PrimExpr()

tvm::tir::IterVar::operator PrimExpr ( ) const
inline
Returns
the corresponding var in the IterVar.

◆ TVM_DEFINE_OBJECT_REF_COW_METHOD()

tvm::tir::IterVar::TVM_DEFINE_OBJECT_REF_COW_METHOD ( IterVarNode  )

◆ TVM_DEFINE_OBJECT_REF_METHODS()

tvm::tir::IterVar::TVM_DEFINE_OBJECT_REF_METHODS ( IterVar  ,
ObjectRef  ,
IterVarNode   
)

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