ONE - On-device Neural Engine
Loading...
Searching...
No Matches
SoftMax.h File Reference
#include "cker/Shape.h"
#include "cker/Utils.h"
#include "cker/Types.h"
#include "cker/eigen/Utils.h"
#include <Eigen/Core>
#include <fixedpoint/fixedpoint.h>
#include <cmath>

Go to the source code of this file.

Namespaces

namespace  nnfw
 
namespace  nnfw::cker
 
namespace  nnfw::cker::reference
 

Functions

void nnfw::cker::reference::Softmax (const SoftmaxParams &params, const Shape &input_shape, const float *input_data, const Shape &output_shape, float *output_data)
 
void nnfw::cker::Softmax (const float *in, const int input_size, const int batch_size, const float beta, float *out)
 
void nnfw::cker::Softmax (const SoftmaxParams &params, const Shape &input_shape, const float *input_data, const Shape &output_shape, float *output_data)
 
template<typename T >
int32_t nnfw::cker::QuantizeSoftmaxOutput (float prob_rescaled, int32_t zero_point)
 
template<>
int32_t nnfw::cker::QuantizeSoftmaxOutput< uint8_t > (float prob_rescaled, int32_t)
 
void nnfw::cker::PopulateSoftmaxLookupTable (float *table, float input_scale, float beta)
 
template<typename In , typename Out >
void nnfw::cker::Softmax (const SoftmaxParams &params, const Shape &input_shape, const In *input_data, const Shape &output_shape, Out *output_data)