#include "cker/TensorUtils.h"
#include "cker/Types.h"

Go to the source code of this file.

Namespaces
namespace	nnfw

namespace	nnfw::cker

Functions
void	nnfw::cker::CalculateLstmGateFloat (const float input, const float input_to_gate_weights, const float aux_input, const float aux_input_to_gate_weights, const float output_state, const float recurrent_to_gate_weights, const float cell_state, const float cell_to_gate_weights, const float layer_norm_coefficients, const float gate_bias, const int n_batch, const int n_input, const int n_aux_input, const int n_output, const int n_cell, const FusedActivationFunctionType activation, float *gate, const bool is_input_all_zeros, const bool is_aux_input_all_zeros)

void	nnfw::cker::UpdateLstmCellFloat (int n_batch, int n_cell, float cell_state, const float input_gate, float forget_gate, const float cell_gate, bool use_cifg, float clip)

void	nnfw::cker::CalculateLstmOutputFloat (int n_batch, int n_cell, int n_output, const float cell_state, const float output_gate, FusedActivationFunctionType activation, const float projection_weights, const float projection_bias, const float proj_clip, float output_state, float scratch)

void	nnfw::cker::LstmStepFloat (const float input_ptr, const float input_to_input_weights_ptr, const float input_to_forget_weights_ptr, const float input_to_cell_weights_ptr, const float input_to_output_weights_ptr, const float aux_input_ptr, const float aux_input_to_input_weights_ptr, const float aux_input_to_forget_weights_ptr, const float aux_input_to_cell_weights_ptr, const float aux_input_to_output_weights_ptr, const float recurrent_to_input_weights_ptr, const float recurrent_to_forget_weights_ptr, const float recurrent_to_cell_weights_ptr, const float recurrent_to_output_weights_ptr, const float cell_to_input_weights_ptr, const float cell_to_forget_weights_ptr, const float cell_to_output_weights_ptr, const float input_layer_norm_coefficients_ptr, const float forget_layer_norm_coefficients_ptr, const float cell_layer_norm_coefficients_ptr, const float output_layer_norm_coefficients_ptr, const float input_gate_bias_ptr, const float forget_gate_bias_ptr, const float cell_gate_bias_ptr, const float output_gate_bias_ptr, const float projection_weights_ptr, const float projection_bias_ptr, const LSTMParams params, int n_batch, int n_cell, int n_input, int n_aux_input, int n_output, int output_batch_leading_dim, float output_state_ptr, float cell_state_ptr, float scratch0, float scratch1, float scratch2, float scratch3, float *output_ptr)

Namespaces

Functions