ONE - On-device Neural Engine
Loading...
Searching...
No Matches
PALSoftmax.h File Reference
#include <tensorflow/lite/kernels/internal/reference/softmax.h>
#include <arm_nnfunctions.h>

Go to the source code of this file.

Namespaces

namespace  luci_interpreter_pal
 

Functions

template<>
void luci_interpreter_pal::Softmax< int8_t > (const tflite::SoftmaxParams &params, const tflite::RuntimeShape &input_shape, const int8_t *input_data, const tflite::RuntimeShape &output_shape, int8_t *output_data)