Function arm_nn_lstm_calculate_gate_s16

• Defined in File arm_nnsupportfunctions.h

Function Documentation

arm_cmsis_nn_status arm_nn_lstm_calculate_gate_s16(const int16_t *data_in, const int16_t *hidden_in, const cmsis_nn_lstm_gate *gate_data, const cmsis_nn_lstm_params *params, int16_t *output, const int32_t batch_offset)

Updates a LSTM gate for an iteration step of LSTM function, int16x8_16 version.

Parameters:

• data_in – [in] Data input pointer

• hidden_in – [in] Hidden state/ recurrent input pointer

• gate_data – [in] Struct containing all information about the gate caluclation, see arm_nn_types.

• params – [in] Struct containing all information about the lstm_operation, see arm_nn_types

• output – [out] Hidden state/ recurrent output pointer

• batch_offset – [in] Number of timesteps between consecutive batches, see arm_nn_lstm_step_s16.

Returns:

The function returns ARM_CMSIS_NN_SUCCESS