ONE - On-device Neural Engine
|
This file contains TopK method and TopContainer class for TopK operation. More...
Go to the source code of this file.
Data Structures | |
class | nnfw::rt::optimized_ops::TopContainer< T > |
class to define TopK operation More... | |
Namespaces | |
namespace | nnfw |
namespace | nnfw::rt |
namespace | nnfw::rt::optimized_ops |
Typedefs | |
typedef int32_t | int32 |
Functions | |
template<typename T > | |
void | nnfw::rt::optimized_ops::TopK (int32 row_size, int32 num_rows, const T *data, int32 k, int32 *output_indexes, T *output_values) |
Operates TopK operation with params. | |
This file contains TopK method and TopContainer class for TopK operation.
Definition in file topk_v2.h.