ONE - On-device Neural Engine
|
#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 ¶ms, const tflite::RuntimeShape &input_shape, const int8_t *input_data, const tflite::RuntimeShape &output_shape, int8_t *output_data) |