tvm
Public Member Functions | Static Public Member Functions | Public Attributes | Static Public Attributes | List of all members
tvm::relax::LayoutTransformAttrs Struct Reference

Attributes used in layout_transform operator. More...

#include <manipulate.h>

Inheritance diagram for tvm::relax::LayoutTransformAttrs:
Collaboration diagram for tvm::relax::LayoutTransformAttrs:

Public Member Functions

 TVM_FFI_DECLARE_FINAL_OBJECT_INFO (LayoutTransformAttrs, BaseAttrsNode)
 
- Public Member Functions inherited from tvm::AttrsNodeReflAdapter< LayoutTransformAttrs >
void InitByPackedArgs (const ffi::PackedArgs &args, bool allow_unknown) final
 Initialize the attributes by arguments. More...
 
- Public Member Functions inherited from tvm::BaseAttrsNode
virtual ~BaseAttrsNode ()
 virtual destructor More...
 
template<typename... Args>
void InitBySeq (Args &&... args)
 Initialize the attributes by sequence of arguments. More...
 
 TVM_DECLARE_BASE_OBJECT_INFO (BaseAttrsNode, Object)
 

Static Public Member Functions

static void RegisterReflection ()
 

Public Attributes

tir::IndexMap index_map
 
Optional< PrimValuepad_value
 
Optional< Array< IntImm > > axis_separators
 
Optional< Array< IntImm > > input_axis_separators
 

Static Public Attributes

static constexpr const char * _type_key = "relax.attrs.LayoutTransformAttrs"
 
- Static Public Attributes inherited from tvm::BaseAttrsNode
static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode
 
static constexpr const char * _type_key = "ir.Attrs"
 

Detailed Description

Attributes used in layout_transform operator.

Member Function Documentation

◆ RegisterReflection()

static void tvm::relax::LayoutTransformAttrs::RegisterReflection ( )
inlinestatic

◆ TVM_FFI_DECLARE_FINAL_OBJECT_INFO()

tvm::relax::LayoutTransformAttrs::TVM_FFI_DECLARE_FINAL_OBJECT_INFO ( LayoutTransformAttrs  ,
BaseAttrsNode   
)

Member Data Documentation

◆ _type_key

constexpr const char* tvm::relax::LayoutTransformAttrs::_type_key = "relax.attrs.LayoutTransformAttrs"
staticconstexpr

◆ axis_separators

Optional<Array<IntImm> > tvm::relax::LayoutTransformAttrs::axis_separators

axis_separators between input axes when generating flattened output axes. For buffers representing flat 1-d memory (e.g. any buffer in RAM), this should be an empty array. For buffers representing non-flat memory, each entry in axis_separators should be the first input axis that is part of a new flattened axis.

◆ index_map

tir::IndexMap tvm::relax::LayoutTransformAttrs::index_map

◆ input_axis_separators

Optional<Array<IntImm> > tvm::relax::LayoutTransformAttrs::input_axis_separators

axis_separators for input buffers. Needed to identify if the input buffer to layout_transform contains axis separator.

◆ pad_value

Optional<PrimValue> tvm::relax::LayoutTransformAttrs::pad_value

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