ONE - On-device Neural Engine
Loading...
Searching...
No Matches
topk_v2.h File Reference

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.
 

Detailed Description

This file contains TopK method and TopContainer class for TopK operation.

Definition in file topk_v2.h.

Typedef Documentation

◆ int32

typedef int32_t int32

Definition at line 27 of file topk_v2.h.