Function arm_argmin_s8

• Defined in File arm_nnfunctions.h

Function Documentation

arm_cmsis_nn_status arm_argmin_s8(const int8_t *input_data, const cmsis_nn_dims *input_dims, const int32_t axis, int32_t *output_data)

Compute ArgMin indices of an s8 tensor along a specific axis.

Parameters:

• input_data – [in] Pointer to input tensor

• input_dims – [in] Input tensor dimensions (NHWC layout)

• axis – [in] Reduction axis in range [0, 3]

• output_data – [out] Pointer to output indices (int32_t)

Returns:

The function returns ARM_CMSIS_NN_SUCCESS for a valid axis, otherwise ARM_CMSIS_NN_ARG_ERROR.