Function arm_elementwise_prelu_s16

Function Documentation

arm_cmsis_nn_status arm_elementwise_prelu_s16(const int16_t *input, const int16_t *alpha, const int32_t input_offset, const int32_t alpha_offset, const int32_t out_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, int16_t *output, const int32_t block_size)

Elementwise S16 PReLU activation function.

Parameters:

  • input – [in] Pointer to the input buffer

  • alpha – [in] Pointer to the alpha buffer (same shape as input)

  • input_offset – [in] Input tensor zero offset

  • alpha_offset – [in] Alpha tensor zero offset

  • out_offset – [in] Output tensor zero offset

  • output_multiplier_identity – [in] Output multiplier when input >= 0

  • output_shift_identity – [in] Output shift when input >= 0

  • output_multiplier_alpha – [in] Output multiplier when input < 0

  • output_shift_alpha – [in] Output shift when input < 0

  • output – [out] Pointer to the output buffer

  • block_size – [in] Number of elements to process

Returns:

ARM_CMSIS_NN_SUCCESS