#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) |
| |