Function arm_nn_lstm_step_s16

• Defined in File arm_nnsupportfunctions.h

Function Documentation

arm_cmsis_nn_status arm_nn_lstm_step_s16(const int16_t *data_in, const int16_t *hidden_in, int16_t *hidden_out, const cmsis_nn_lstm_params *params, cmsis_nn_lstm_context *buffers, const int32_t batch_offset)

Update LSTM function for an iteration step using s16 input and output, and s16 internally.

Parameters:

• data_in – [in] Data input pointer

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

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

• params – [in] Struct containg all information about the lstm operator, see arm_nn_types.

• buffers – [in] Struct containg pointers to all temporary scratch buffers needed for the lstm operator, see arm_nn_types.

• batch_offset – [in] Number of timesteps between consecutive batches. E.g for params->timing_major = true, all batches for t=0 are stored sequentially, so batch offset = 1. For params->time major = false, all time steps are stored continously before the next batch, so batch offset = params->time_steps.

Returns:

The function returns ARM_CMSIS_NN_SUCCESS