Attributes used in batch_norm operator. More...

#include <nn.h>

Inheritance diagram for tvm::relax::BatchNormAttrs:

Collaboration diagram for tvm::relax::BatchNormAttrs:

Public Member Functions
	TVM_FFI_DECLARE_FINAL_OBJECT_INFO (BatchNormAttrs, BaseAttrsNode)

Public Member Functions inherited from tvm::AttrsNodeReflAdapter< BatchNormAttrs >
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
int	axis

double	epsilon

bool	center

bool	scale

double	momentum

bool	training

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

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 batch_norm operator.

Member Function Documentation

◆ RegisterReflection()

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

inlinestatic

◆ TVM_FFI_DECLARE_FINAL_OBJECT_INFO()

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

Member Data Documentation

◆ _type_key

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

staticconstexpr

◆ axis

int tvm::relax::BatchNormAttrs::axis

◆ center

bool tvm::relax::BatchNormAttrs::center

◆ epsilon

double tvm::relax::BatchNormAttrs::epsilon

◆ momentum

double tvm::relax::BatchNormAttrs::momentum

◆ scale

bool tvm::relax::BatchNormAttrs::scale

◆ training

bool tvm::relax::BatchNormAttrs::training

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