tvm::relax::LayerNormAttrs Struct Reference

Attributes used in layer_norm operator. More...

#include <nn.h>

Inheritance diagram for tvm::relax::LayerNormAttrs:

Collaboration diagram for tvm::relax::LayerNormAttrs:

Public Member Functions
	TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("relax.attrs.LayerNormAttrs", LayerNormAttrs, BaseAttrsNode)
Public Member Functions inherited from tvm::AttrsNodeReflAdapter< LayerNormAttrs >
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< Integer >	axes
double	epsilon
bool	center
bool	scale

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

Detailed Description

Attributes used in layer_norm operator.

Member Function Documentation

RegisterReflection()

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

inlinestatic

TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

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

Member Data Documentation

axes

ffi::Array<Integer> tvm::relax::LayerNormAttrs::axes

center

bool tvm::relax::LayerNormAttrs::center

epsilon

double tvm::relax::LayerNormAttrs::epsilon

scale

bool tvm::relax::LayerNormAttrs::scale

---

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

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