Function arm_nn_lstm_step_f16

• Defined in File arm_nnsupportfunctions_flt.h

Function Documentation

arm_cmsis_nn_status arm_nn_lstm_step_f16(const float16_t *data_in, const float16_t *hidden_in, float16_t *hidden_out, const cmsis_nn_lstm_params_f16 *params, cmsis_nn_lstm_context_f16 *buffers, const int32_t batch_offset)

Update LSTM function for an iteration step using float16 input, output and state.

Parameters:

• data_in – [in] Data input pointer.

• hidden_in – [in] Hidden state / recurrent input pointer. May be NULL for the first step.

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

• params – [in] Struct containing all information about the LSTM operator.

• buffers – [in] Struct containing pointers to mutable cell-state storage.

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

Returns:

The function returns ARM_CMSIS_NN_SUCCESS.