|
tvm
|
#include <layout.h>
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. | |
| template<typename ValueType > | |
| static AxisAttrMap< ValueType > | GetAttrMap (const ffi::String &attr_name) |
| Get the attribute map for the axis. | |
Friends | |
| template<typename , typename > | |
| class | tvm::AttrRegistry |
| class | AxisRegEntry |
|
default |
|
inlineexplicit |
|
inlinestatic |
Get the attribute map for the axis.
|
friend |
|
friend |