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_OBJECT_INFO_FINAL ("relax.attrs.Pool1DAttrs", Pool1DAttrs, AttrsNode)
Public Member Functions inherited from tvm::AttrsNode
	TVM_FFI_DECLARE_OBJECT_INFO ("ir.Attrs", AttrsNode, ffi::Object)

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
ffi::Array< int64_t >	pool_size
ffi::Array< int64_t >	strides
ffi::Array< int64_t >	padding
ffi::Array< int64_t >	dilation
bool	ceil_mode
bool	count_include_pad
ffi::String	layout
ffi::String	out_layout

Additional Inherited Members
Static Public Attributes inherited from tvm::AttrsNode
static constexpr TVMFFISEqHashKind	_type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode

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_OBJECT_INFO_FINAL()

tvm::relax::Pool1DAttrs::TVM_FFI_DECLARE_OBJECT_INFO_FINAL	(	"relax.attrs.Pool1DAttrs"	,
		Pool1DAttrs	,
		AttrsNode
	)

Member Data Documentation

ceil_mode

bool tvm::relax::Pool1DAttrs::ceil_mode

count_include_pad

bool tvm::relax::Pool1DAttrs::count_include_pad

dilation

ffi::Array<int64_t> tvm::relax::Pool1DAttrs::dilation

layout

ffi::String tvm::relax::Pool1DAttrs::layout

out_layout

ffi::String tvm::relax::Pool1DAttrs::out_layout

padding

ffi::Array<int64_t> tvm::relax::Pool1DAttrs::padding

pool_size

ffi::Array<int64_t> tvm::relax::Pool1DAttrs::pool_size

strides

ffi::Array<int64_t> tvm::relax::Pool1DAttrs::strides

---

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

• include/tvm/relax/attrs/nn.h