node container for IterVar attr More...

#include <schedule.h>

Inheritance diagram for tvm::te::IterVarAttrNode:

Collaboration diagram for tvm::te::IterVarAttrNode:

Public Member Functions
void	VisitAttrs (AttrVisitor *v)

	TVM_DECLARE_FINAL_OBJECT_INFO (IterVarAttrNode, Object)

Public Member Functions inherited from tvm::runtime::Object
uint32_t	type_index () const

std::string	GetTypeKey () const

size_t	GetTypeKeyHash () const

template<typename TargetType >
bool	IsInstance () const

bool	unique () const

	Object ()

	Object (const Object &other)

	Object (Object &&other)

Object &	operator= (const Object &other)

Object &	operator= (Object &&other)

Public Attributes
IterVarType	iter_type {kDataPar}
	The iteration type. More...

IterVar	bind_thread
	The thread this iter Var binds, can be null. More...

Array< Tensor >	prefetch_data
	List of tensor to be prefetched in this loop. More...

Array< PrimExpr >	prefetch_offset
	The offset used in each prefetch. More...

TensorIntrin	tensor_intrin
	Tensor intrinsic used in tensorization, when the axis is marked as Tensorized. More...

int	dim_align_factor {0}
	Alignment factor of buffer dimension. More...

int	dim_align_offset {0}
	Alignment offset of buffer dimension. More...

Array< PrimExpr >	pragma_keys
	Additional pragma keys, array of StringImm. More...

Array< PrimExpr >	pragma_values
	Additional values of pragma, if any. More...

Static Public Attributes
static constexpr const char *	_type_key = "IterVarAttr"

Static Public Attributes inherited from tvm::runtime::Object
static constexpr const char *	_type_key = "runtime.Object"

static constexpr bool	_type_final = false

static constexpr uint32_t	_type_child_slots = 0

static constexpr bool	_type_child_slots_can_overflow = true

static constexpr bool	_type_has_method_visit_attrs = true

static constexpr bool	_type_has_method_sequal_reduce = false

static constexpr bool	_type_has_method_shash_reduce = false

static constexpr uint32_t	_type_index = TypeIndex::kDynamic

Additional Inherited Members
Public Types inherited from tvm::runtime::Object
typedef void(*	FDeleter) (Object *self)
	Object deleter. More...

using	RefCounterType = std::atomic< int32_t >

Static Public Member Functions inherited from tvm::runtime::Object
static std::string	TypeIndex2Key (uint32_t tindex)
	Get the type key of the corresponding index from runtime. More...

static size_t	TypeIndex2KeyHash (uint32_t tindex)
	Get the type key hash of the corresponding index from runtime. More...

static uint32_t	TypeKey2Index (const std::string &key)
	Get the type index of the corresponding key from runtime. More...

static uint32_t	_GetOrAllocRuntimeTypeIndex ()

static uint32_t	RuntimeTypeIndex ()

Protected Member Functions inherited from tvm::runtime::Object
void	IncRef ()
	developer function, increases reference counter. More...

void	DecRef ()
	developer function, decrease reference counter. More...

Static Protected Member Functions inherited from tvm::runtime::Object
static uint32_t	GetOrAllocRuntimeTypeIndex (const std::string &key, uint32_t static_tindex, uint32_t parent_tindex, uint32_t type_child_slots, bool type_child_slots_can_overflow)
	Get the type index using type key. More...

Protected Attributes inherited from tvm::runtime::Object
uint32_t	type_index_ {0}
	Type index(tag) that indicates the type of the object. More...

RefCounterType	ref_counter_ {0}
	The internal reference counter. More...

FDeleter	deleter_ = nullptr
	deleter of this object to enable customized allocation. If the deleter is nullptr, no deletion will be performed. The creator of the object must always set the deleter field properly. More...

Detailed Description

node container for IterVar attr

Member Function Documentation

◆ TVM_DECLARE_FINAL_OBJECT_INFO()

tvm::te::IterVarAttrNode::TVM_DECLARE_FINAL_OBJECT_INFO	(	IterVarAttrNode	,
		Object
	)

◆ VisitAttrs()

void tvm::te::IterVarAttrNode::VisitAttrs ( AttrVisitor * v )

inline

Member Data Documentation

◆ _type_key

constexpr const char* tvm::te::IterVarAttrNode::_type_key = "IterVarAttr"

staticconstexpr

◆ bind_thread

IterVar tvm::te::IterVarAttrNode::bind_thread

The thread this iter Var binds, can be null.

◆ dim_align_factor

int tvm::te::IterVarAttrNode::dim_align_factor {0}

Alignment factor of buffer dimension.

◆ dim_align_offset

int tvm::te::IterVarAttrNode::dim_align_offset {0}

Alignment offset of buffer dimension.

◆ iter_type

IterVarType tvm::te::IterVarAttrNode::iter_type {kDataPar}

The iteration type.

◆ pragma_keys

Array<PrimExpr> tvm::te::IterVarAttrNode::pragma_keys

Additional pragma keys, array of StringImm.

◆ pragma_values

Array<PrimExpr> tvm::te::IterVarAttrNode::pragma_values

Additional values of pragma, if any.

◆ prefetch_data

Array<Tensor> tvm::te::IterVarAttrNode::prefetch_data

List of tensor to be prefetched in this loop.

◆ prefetch_offset

Array<PrimExpr> tvm::te::IterVarAttrNode::prefetch_offset

The offset used in each prefetch.

◆ tensor_intrin

TensorIntrin tvm::te::IterVarAttrNode::tensor_intrin

Tensor intrinsic used in tensorization, when the axis is marked as Tensorized.

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

include/tvm/te/schedule.h

Public Member Functions

Public Attributes

Static Public Attributes

Additional Inherited Members