ONE - On-device Neural Engine
Loading...
Searching...
No Matches
PALSoftmaxCommon.h File Reference
#include "core/OMKernelData.h"
#include <cmath>

Go to the source code of this file.

Namespaces

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

Functions

template<typename T , typename U >
OMStatus onert_micro::execute::pal::Softmax (const core::SoftmaxParams &params, const T *input_data, U *output_data)
 
template<>
OMStatus onert_micro::execute::pal::Softmax< float, float > (const core::SoftmaxParams &params, const float *input_data, float *output_data)