Function arm_comparison_s16

Function Documentation

arm_cmsis_nn_status arm_comparison_s16(const cmsis_nn_context *ctx, const int16_t *input_1_data, const cmsis_nn_dims *input_1_dims, const int16_t *input_2_data, const cmsis_nn_dims *input_2_dims, bool *output_data, const cmsis_nn_dims *output_dims, const int32_t input_1_offset, const int32_t input_1_mult, const int32_t input_1_shift, const int32_t input_2_offset, const int32_t input_2_mult, const int32_t input_2_shift, const int32_t left_shift, arm_nn_compare_operation operation)

s16 elementwise comparison with support for broadcasting.

Parameters:

  • ctx – [in] Temporary scratch buffer The caller is expected to clear the buffer, if applicable, for security reasons.

  • input_1_data – [in] Pointer to input1 tensor

  • input_1_dims – [in] Input1 tensor dimensions

  • input_2_data – [in] Pointer to input2 tensor

  • input_2_dims – [in] Input2 tensor dimensions

  • output_data – [out] Pointer to the output tensor (bool values)

  • output_dims – [in] Output tensor dimensions

  • input_1_offset – [in] Zero-point for input1 tensor

  • input_1_mult – [in] Multiplier for input1 tensor

  • input_1_shift – [in] Shift for input1 tensor

  • input_2_offset – [in] Zero-point for input2 tensor

  • input_2_mult – [in] Multiplier for input2 tensor

  • input_2_shift – [in] Shift for input2 tensor

  • left_shift – [in] Common left shift prior to requantization

  • operation – [in] Comparison operation to perform

Returns:

The function returns ARM_CMSIS_NN_SUCCESS