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

Go to the source code of this file.

Namespaces

namespace  onert_micro
 
namespace  onert_micro::execute
 
namespace  onert_micro::execute::pal
 

Functions

template<typename T1 , typename T2 , typename T3 >
OMStatus onert_micro::execute::pal::ArgMax (const core::OMRuntimeShape &input1_shape, const T1 *input1_data, const T3 *input2_data, const core::OMRuntimeShape &output_shape, T2 *output_data)