Function arm_lstm_unidirectional_s16_temp1_get_buffer_size

• Defined in File arm_nnfunctions.h

Function Documentation

int32_t arm_lstm_unidirectional_s16_temp1_get_buffer_size(const cmsis_nn_lstm_params *lstm_params)

Get size of the temp1 scratch buffer required by arm_lstm_unidirectional_s16(). Refer to arm_lstm_unidirectional_s8_temp1_get_buffer_size() for argument details, the -1-on-invalid contract and the time_steps / degenerate-0 notes.

Returns:

Required buffer size in bytes: (time_major != 0 ? batch_size : 1) * hidden_size * sizeof(int16_t) - the same figure as arm_lstm_unidirectional_s8_temp1_get_buffer_size() for the same params, since both layer datatypes stage int16_t gate vectors.