Function arm_nn_gru_step_f16

Function Documentation

arm_cmsis_nn_status arm_nn_gru_step_f16(const float16_t *data_in, const float16_t *hidden_in, float16_t *hidden_out, const cmsis_nn_gru_params_f16 *params, cmsis_nn_gru_context_f16 *buffers, const int32_t batch_offset)

Update GRU function for a single iteration step using float16 data.

Parameters:

  • data_in – [in] Data input pointer for this time step.

  • hidden_in – [in] Recurrent input pointer. NULL for the first step (h_prev = 0).

  • hidden_out – [out] Hidden-state output pointer for this time step.

  • params – [in] Struct describing the GRU operator.

  • buffers – [inout] Scratch buffers. temp1 (>= hidden_size) is required when reset_after == 0.

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

Returns:

ARM_CMSIS_NN_SUCCESS on success, or ARM_CMSIS_NN_ARG_ERROR on invalid arguments (NULL data_in/hidden_out/params, batch_offset <= 0, or missing temp1 when reset_after == 0).