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

Go to the source code of this file.

Namespaces

namespace  nnfw
 
namespace  nnfw::cker
 

Functions

template<typename T1 , typename T2 , typename Cmp >
void nnfw::cker::ArgMinMax (const Shape &input1_shape, const T1 *input1_data, const Shape &output_shape, T2 *output_data, int32_t axis, const Cmp &cmp)