Function arm_lstm_unidirectional_s8

Function Documentation

arm_cmsis_nn_status arm_lstm_unidirectional_s8(const int8_t *input, int8_t *output, const cmsis_nn_lstm_params *params, cmsis_nn_lstm_context *buffers)

LSTM unidirectional function with 8 bit input and output and 16 bit gate output, 32 bit bias.

  1. Supported framework: TensorFlow Lite Micro

Parameters:

  • input – [in] Pointer to input data

  • output – [out] Pointer to output data

  • params – [in] Struct containing all information about the lstm operator, see arm_nn_types.

  • buffers – [in] Struct containing pointers to all temporary scratch buffers needed for the lstm operator, see arm_nn_types. Size temp1 with arm_lstm_unidirectional_s8_temp1_get_buffer_size() and temp2 with arm_lstm_unidirectional_s8_temp2_get_buffer_size() - both hold int16_t gate vectors even though the layer datatype is s8, so sizing them in s8 elements under-allocates by half.

Returns:

The function returns ARM_CMSIS_NN_SUCCESS