tvm::relax::TopKAttrs Struct Reference

Attributes used in topk operator. More...

#include <sorting.h>

Inheritance diagram for tvm::relax::TopKAttrs:

Collaboration diagram for tvm::relax::TopKAttrs:

Public Member Functions
	TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("relax.attrs.TopKAttrs", TopKAttrs, AttrsNode)
Public Member Functions inherited from tvm::AttrsNode
	TVM_FFI_DECLARE_OBJECT_INFO ("ir.Attrs", AttrsNode, ffi::Object)

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
int	k
int	axis
bool	largest
ffi::String	ret_type
ffi::Optional< DLDataType >	dtype

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

Detailed Description

Attributes used in topk operator.

Member Function Documentation

RegisterReflection()

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

inlinestatic

TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::relax::TopKAttrs::TVM_FFI_DECLARE_OBJECT_INFO_FINAL	(	"relax.attrs.TopKAttrs"	,
		TopKAttrs	,
		AttrsNode
	)

Member Data Documentation

axis

int tvm::relax::TopKAttrs::axis

dtype

ffi::Optional<DLDataType> tvm::relax::TopKAttrs::dtype

k

int tvm::relax::TopKAttrs::k

largest

bool tvm::relax::TopKAttrs::largest

ret_type

ffi::String tvm::relax::TopKAttrs::ret_type

---

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

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