Attributes used in max_pool2d and avg_pool2d operator. More...

#include <nn.h>

Inheritance diagram for tvm::relax::Pool2DAttrs:

Collaboration diagram for tvm::relax::Pool2DAttrs:

Public Member Functions
	TVM_FFI_DECLARE_FINAL_OBJECT_INFO (Pool2DAttrs, BaseAttrsNode)

Public Member Functions inherited from tvm::AttrsNodeReflAdapter< Pool2DAttrs >
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< IntImm >	pool_size

Array< IntImm >	strides

Array< IntImm >	padding

Array< IntImm >	dilation

bool	ceil_mode

bool	count_include_pad

String	layout

String	out_layout

Static Public Attributes
static constexpr const char *	_type_key = "relax.attrs.Pool2DAttrs"

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_pool2d and avg_pool2d operator.

Member Function Documentation

◆ RegisterReflection()

static void tvm::relax::Pool2DAttrs::RegisterReflection ( )

inlinestatic

◆ TVM_FFI_DECLARE_FINAL_OBJECT_INFO()

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

Member Data Documentation

◆ _type_key

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

staticconstexpr

◆ ceil_mode

bool tvm::relax::Pool2DAttrs::ceil_mode

◆ count_include_pad

bool tvm::relax::Pool2DAttrs::count_include_pad

◆ dilation

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

◆ layout

String tvm::relax::Pool2DAttrs::layout

◆ out_layout

String tvm::relax::Pool2DAttrs::out_layout

◆ padding

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

◆ pool_size

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

◆ strides

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

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

include/tvm/relax/attrs/nn.h

Public Member Functions

Static Public Member Functions

Public Attributes

Static Public Attributes