Function arm_softmax_u8

Function Documentation

void arm_softmax_u8(const uint8_t *input, const int32_t num_rows, const int32_t row_size, const int32_t mult, const int32_t shift, const int32_t diff_min, uint8_t *output)

U8 softmax function.

Note

Supported framework: TensorFlow Lite micro (bit-accurate)

Parameters:

  • input – [in] Pointer to the input tensor

  • num_rows – [in] Number of rows in the input tensor

  • row_size – [in] Number of elements in each input row

  • mult – [in] Input quantization multiplier

  • shift – [in] Input quantization shift within the range [0, 31]

  • diff_min – [in] Minimum difference with max in row. Used to check if the quantized exponential operation can be performed

  • output – [out] Pointer to the output tensor