ONE - On-device Neural Engine
Loading...
Searching...
No Matches
PALGRU.h File Reference
#include <tensorflow/lite/kernels/internal/reference/fully_connected.h>
#include "PALreference_ops.h"

Go to the source code of this file.

Namespaces

namespace  luci_interpreter_pal
 

Functions

void luci_interpreter_pal::Logistic (const int flat_size, const float *input_data, float *output_data)
 
void luci_interpreter_pal::calculateGRU (const float *input_data, const float *weight_input_data, const float *weight_hidden_data, const float *bias_input_data, const float *bias_hidden_data, float *output_data, const tflite::RuntimeShape &input_shape, const tflite::RuntimeShape &output_shape, const tflite::RuntimeShape &weight_input_shape, const tflite::RuntimeShape &weight_hidden_shape, float *output_input_data, float *output_hidden_data, const tflite::RuntimeShape &output_shape_fc)
 
void luci_interpreter_pal::GRU (const float *input_data, const float *weight_input_data, const float *weight_hidden_data, const float *bias_input_data, const float *bias_hidden_data, const float *hidden_state_data, float *output_data, float *output_input_data, float *output_hidden_data, const tflite::RuntimeShape &input_shape, const tflite::RuntimeShape &output_shape, const tflite::RuntimeShape &weight_input_shape, const tflite::RuntimeShape &weight_hidden_shape)