tvm::relax::BatchNormAttrs Struct Reference

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_OBJECT_INFO_FINAL ("relax.attrs.BatchNormAttrs", 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_FFI_DECLARE_OBJECT_INFO ("ir.Attrs", BaseAttrsNode, Object)

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
int	axis
double	epsilon
bool	center
bool	scale
double	momentum
bool	training

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

Detailed Description

Attributes used in batch_norm operator.

Member Function Documentation

RegisterReflection()

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

inlinestatic

TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

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

Member Data Documentation

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