Struct cmsis_nn_lstm_params_f16

• Defined in File arm_nn_types_flt.h

Struct Documentation

struct cmsis_nn_lstm_params_f16

Parameters for a unidirectional float32 LSTM layer.

Public Members

int32_t time_major

Non-zero when input/output tensors are time-major.

int32_t batch_size

Batch size processed per invocation.

int32_t time_steps

Number of time steps processed per invocation.

int32_t input_size

Input feature size per time step.

int32_t hidden_size

Hidden-state size.

float16_t cell_clip

Optional cell-state clip value.

cmsis_nn_lstm_gate_f16 forget_gate

Forget gate weights and activation.

cmsis_nn_lstm_gate_f16 input_gate

Input gate weights and activation.

cmsis_nn_lstm_gate_f16 cell_gate

Cell-update gate weights and activation.

cmsis_nn_lstm_gate_f16 output_gate

Output gate weights and activation.