ONE - On-device Neural Engine
Loading...
Searching...
No Matches
LogSoftMax.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
 

Functions

void nnfw::cker::LogSoftmax (const SoftmaxParams &params, const Shape &input_shape, const float *input_data, const Shape &output_shape, float *output_data)
 
void nnfw::cker::LogSoftmax (const SoftmaxParams &params, float input_scale, const Shape &input_shape, const uint8_t *input_data, const Shape &output_shape, uint8_t *output_data)