Function arm_where_s16¶

Defined in File arm_nnfunctions.h

Function Documentation¶

arm_cmsis_nn_status arm_where_s16(const int16_t *condition, const cmsis_nn_where_params *params, int64_t *output, int32_t *num_true)¶

WHERE operator: return coordinates of non-zero elements in condition (int16).

Parameters:

condition – [in] Pointer to condition tensor data (int16, non-zero = true)
params – [in] Pointer to where parameters (rank, shape)
output – [out] Pointer to output coordinates (int64, shape [max_true, rank])
num_true – [out] Number of true elements found

Returns:

The function returns ARM_CMSIS_NN_SUCCESS