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.
-
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
-
inline TopKFusion()