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