Function arm_softmax_f16

• Defined in File arm_nnfunctions_flt.h

Function Documentation

arm_cmsis_nn_status arm_softmax_f16(const float16_t *input, int32_t num_rows, int32_t row_size, float16_t *output)

Softmax using the float-native API signature.

Parameters:

• input – [in] Pointer to the input matrix stored as num_rows rows of row_size values.

• num_rows – [in] Number of rows in the input matrix.

• row_size – [in] Number of columns per row.

• output – [out] Pointer to the output matrix.

Returns:

ARM_CMSIS_NN_SUCCESS on success or ARM_CMSIS_NN_ARG_ERROR on invalid arguments.