#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.
|
void | nnfw::cker::reference::Softmax (const SoftmaxParams ¶ms, 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 ¶ms, 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 ¶ms, const Shape &input_shape, const In *input_data, const Shape &output_shape, Out *output_data) |
|