tvm
Public Member Functions | Static Public Member Functions | Friends | List of all members
tvm::tirx::Axis Class Reference

#include <layout.h>

Inheritance diagram for tvm::tirx::Axis:
Collaboration diagram for tvm::tirx::Axis:

Public Member Functions

 Axis ()=default
 
 Axis (ffi::ObjectPtr< AxisNode > data)
 
 TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE (Axis, ffi::ObjectRef, AxisNode)
 

Static Public Member Functions

static Axis Get (const ffi::String &name)
 Get the axis object by name. More...
 
template<typename ValueType >
static AxisAttrMap< ValueType > GetAttrMap (const ffi::String &attr_name)
 Get the attribute map for the axis. More...
 

Friends

template<typename , typename >
class tvm::AttrRegistry
 
class AxisRegEntry
 

Constructor & Destructor Documentation

◆ Axis() [1/2]

tvm::tirx::Axis::Axis ( )
default

◆ Axis() [2/2]

tvm::tirx::Axis::Axis ( ffi::ObjectPtr< AxisNode data)
inlineexplicit

Member Function Documentation

◆ Get()

static Axis tvm::tirx::Axis::Get ( const ffi::String &  name)
static

Get the axis object by name.

◆ GetAttrMap()

template<typename ValueType >
static AxisAttrMap<ValueType> tvm::tirx::Axis::GetAttrMap ( const ffi::String &  attr_name)
inlinestatic

Get the attribute map for the axis.

◆ TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE()

tvm::tirx::Axis::TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE ( Axis  ,
ffi::ObjectRef  ,
AxisNode   
)

Friends And Related Function Documentation

◆ AxisRegEntry

friend class AxisRegEntry
friend

◆ tvm::AttrRegistry

template<typename , typename >
friend class tvm::AttrRegistry
friend

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