Function arm_prelu_s8¶
Defined in File arm_nnfunctions.h
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