Class TopKFusion

Inheritance Relationships

class TopKFusion : public mindspore::ops::TopK 

TopKFusion defined TopK operator prototype of lite.

Public Functions

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

Method to init the op's attributes.

参数

sorted – [in] Define a boolean value indicate whether the output must 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.

参数: 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.

参数: 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.

int64_t get_largest() const

Method to get largest attribute.