Attributes used in Conv2d operator. More...

#include <nn.h>

Inheritance diagram for tvm::relax::Conv2DAttrs:

Collaboration diagram for tvm::relax::Conv2DAttrs:

Public Member Functions
	TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("relax.attrs.Conv2DAttrs", Conv2DAttrs, BaseAttrsNode)

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

ffi::Array< IntImm >	padding

ffi::Array< IntImm >	dilation

int	groups

ffi::String	data_layout

ffi::String	kernel_layout

ffi::String	out_layout

DataType	out_dtype

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

Detailed Description

Attributes used in Conv2d operator.

Member Function Documentation

◆ RegisterReflection()

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

inlinestatic

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

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

Member Data Documentation

◆ data_layout

ffi::String tvm::relax::Conv2DAttrs::data_layout

◆ dilation

ffi::Array<IntImm> tvm::relax::Conv2DAttrs::dilation

◆ groups

int tvm::relax::Conv2DAttrs::groups

◆ kernel_layout

ffi::String tvm::relax::Conv2DAttrs::kernel_layout

◆ out_dtype

DataType tvm::relax::Conv2DAttrs::out_dtype

◆ out_layout

ffi::String tvm::relax::Conv2DAttrs::out_layout

◆ padding

ffi::Array<IntImm> tvm::relax::Conv2DAttrs::padding

◆ strides

ffi::Array<IntImm> tvm::relax::Conv2DAttrs::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

Additional Inherited Members