ONE - On-device Neural Engine
|
#include "PALArgMinMaxCommon.h"
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) |