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

Attributes used in Conv1d operator. More...

#include <nn.h>

Inheritance diagram for tvm::relax::Conv1DAttrs:
Collaboration diagram for tvm::relax::Conv1DAttrs:

Public Member Functions

 TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("relax.attrs.Conv1DAttrs", Conv1DAttrs, BaseAttrsNode)
 
- Public Member Functions inherited from tvm::AttrsNodeReflAdapter< Conv1DAttrs >
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_FFI_DECLARE_OBJECT_INFO ("ir.Attrs", BaseAttrsNode, Object)
 

Static Public Member Functions

static void RegisterReflection ()
 

Public Attributes

ffi::Array< IntImmstrides
 
ffi::Array< IntImmpadding
 
ffi::Array< IntImmdilation
 
int groups
 
ffi::String data_layout
 
ffi::String kernel_layout
 
ffi::String out_layout
 
DataType out_dtype
 

Additional Inherited Members

- Static Public Attributes inherited from tvm::BaseAttrsNode
static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode
 

Detailed Description

Attributes used in Conv1d operator.

Member Function Documentation

◆ RegisterReflection()

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

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::relax::Conv1DAttrs::TVM_FFI_DECLARE_OBJECT_INFO_FINAL ( "relax.attrs.Conv1DAttrs"  ,
Conv1DAttrs  ,
BaseAttrsNode   
)

Member Data Documentation

◆ data_layout

ffi::String tvm::relax::Conv1DAttrs::data_layout

◆ dilation

ffi::Array<IntImm> tvm::relax::Conv1DAttrs::dilation

◆ groups

int tvm::relax::Conv1DAttrs::groups

◆ kernel_layout

ffi::String tvm::relax::Conv1DAttrs::kernel_layout

◆ out_dtype

DataType tvm::relax::Conv1DAttrs::out_dtype

◆ out_layout

ffi::String tvm::relax::Conv1DAttrs::out_layout

◆ padding

ffi::Array<IntImm> tvm::relax::Conv1DAttrs::padding

◆ strides

ffi::Array<IntImm> tvm::relax::Conv1DAttrs::strides

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