ONE - On-device Neural Engine
Loading...
Searching...
No Matches
TopKV2.h File Reference
#include "cker/Shape.h"

Go to the source code of this file.

Data Structures

class  nnfw::cker::TopContainer< T, Tidx >
 

Namespaces

namespace  nnfw
 
namespace  nnfw::cker
 

Functions

template<typename T , typename Tidx = int32_t>
void nnfw::cker::TopKV2 (const Shape &input_shape, const T *input_data, const uint32_t k, T *output_value_data, Tidx *output_indices_data)