Function arm_prelu_s8

Function Documentation

arm_cmsis_nn_status arm_prelu_s8(const cmsis_nn_dims *input_dims, const int8_t *input, const cmsis_nn_dims *alpha_dims, const int8_t *alpha, const int32_t input_offset, const int32_t alpha_offset, const int32_t output_offset, const int32_t output_multiplier_identity, const int32_t output_shift_identity, const int32_t output_multiplier_alpha, const int32_t output_shift_alpha, const cmsis_nn_dims *output_dims, int8_t *output)

S8 PReLU activation function.

Parameters:
  • input_dims[in] Input (activation) tensor dimensions. Format: [N, H, W, C_IN]

  • input[in] Pointer to the input buffer

  • alpha_dims[in] Alpha tensor dimensions. Format: [N, H, W, C]

  • alpha[in] Pointer to the alpha buffer

  • input_offset[in] Input tensor zero offset

  • alpha_offset[in] Alpha tensor zero offset

  • output_offset[in] Output tensor zero offset

  • output_multiplier_identity[in] Output multiplier 1

  • output_shift_identity[in] Output shift 1

  • output_multiplier_alpha[in] Output multiplier 2

  • output_shift_alpha[in] Output shift 2

  • output_dims[in] Output tensor dimensions. Format: [N, H, W, C_OUT]

  • output[out] Pointer to the output buffer

Returns:

The function returns ARM_MATH_SUCCESS