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

Attributes used in max_pool1d and avg_pool1d operator. More...

#include <nn.h>

Inheritance diagram for tvm::relax::Pool1DAttrs:
Collaboration diagram for tvm::relax::Pool1DAttrs:

Public Member Functions

 TVM_FFI_DECLARE_FINAL_OBJECT_INFO (Pool1DAttrs, BaseAttrsNode)
 
- Public Member Functions inherited from tvm::AttrsNodeReflAdapter< Pool1DAttrs >
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

Array< IntImmpool_size
 
Array< IntImmstrides
 
Array< IntImmpadding
 
Array< IntImmdilation
 
bool ceil_mode
 
bool count_include_pad
 
String layout
 
String out_layout
 

Static Public Attributes

static constexpr const char * _type_key = "relax.attrs.Pool1DAttrs"
 
- 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 max_pool1d and avg_pool1d operator.

Member Function Documentation

◆ RegisterReflection()

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

◆ TVM_FFI_DECLARE_FINAL_OBJECT_INFO()

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

Member Data Documentation

◆ _type_key

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

◆ ceil_mode

bool tvm::relax::Pool1DAttrs::ceil_mode

◆ count_include_pad

bool tvm::relax::Pool1DAttrs::count_include_pad

◆ dilation

Array<IntImm> tvm::relax::Pool1DAttrs::dilation

◆ layout

String tvm::relax::Pool1DAttrs::layout

◆ out_layout

String tvm::relax::Pool1DAttrs::out_layout

◆ padding

Array<IntImm> tvm::relax::Pool1DAttrs::padding

◆ pool_size

Array<IntImm> tvm::relax::Pool1DAttrs::pool_size

◆ strides

Array<IntImm> tvm::relax::Pool1DAttrs::strides

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