Function arm_lstm_unidirectional_s8_temp2_get_buffer_size

• Defined in File arm_nnfunctions.h

Function Documentation

int32_t arm_lstm_unidirectional_s8_temp2_get_buffer_size(const cmsis_nn_lstm_params *lstm_params)

Get size of the temp2 scratch buffer required by arm_lstm_unidirectional_s8(). Refer to arm_lstm_unidirectional_s8_temp1_get_buffer_size() for argument details and the -1-on-invalid contract.

Returns:

Required buffer size in bytes: the same figure as arm_lstm_unidirectional_s8_temp1_get_buffer_size() for the same params. temp2 stages the cell-gate vector and the tanh(cell_state) vector, both of the same extent as the gate vectors staged in temp1.