Class TopKFusion

Inheritance Relationships

Base Type

Class Documentation

class TopKFusion : public mindspore::ops::TopK

TopKFusion defined TopK operator prototype of lite.

Public Functions

inline TopKFusion()

Constructor.

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 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.

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