Class TopKFusion

• Defined in File topk_fusion.h

Inheritance Relationships

Base Type

• public mindspore::ops::TopK (Class TopK)

Class Documentation

class TopKFusion : public mindspore::ops::TopK

TopKFusion defined TopK operator prototype of lite.

Public Functions

inline TopKFusion()

Constructor.

~TopKFusion() = default

Destructor.

void Init(const bool sorted, const int64_t axis, const int64_t largest)

Method to init the op’s attributes.

Parameters

• sorted – [in] Define a boolean value indicate whether the output should be sorted.

• axis – [in] Define the operation axis, which is no use now, but the default is the last dimension.

• largest – [in] Define the number of largest value along with the axis, which is not attribute now, but the second input of this operation.

void set_axis(const int64_t axis)

Method to set axis attribute. Do not use.

Parameters

axis – [in] Define the operation axis, which is no use now, but the default is the last dimension.

void set_largest(const int64_t largest)

Method to set largest attribute, which is no use and needs to be converted to the second input.

Parameters

largest – [in] Define the number of largest value along with the axis, which is not attribute now, but the second input of this operation.

int64_t get_axis() const

Method to get axis attribute.

Returns

axis value.

int64_t get_largest() const

Method to get largest attribute.

Returns

the number of largest value