Function arm_prelu_f16

Function Documentation

arm_cmsis_nn_status arm_prelu_f16(const cmsis_nn_dims *input_dims, const float16_t *input, const cmsis_nn_dims *alpha_dims, const float16_t *alpha, const cmsis_nn_dims *output_dims, float16_t *output)

Parametric ReLU for float32 data.

Computes output = input >= 0 ? input : input * alpha, with alpha broadcast onto the input (TensorFlow Lite semantics): each alpha dimension must equal the matching input dimension or 1.

Parameters:

  • input_dims – [in] Input tensor dimensions. Must equal output_dims.

  • input – [in] Pointer to the input tensor.

  • alpha_dims – [in] Alpha tensor dimensions.

  • alpha – [in] Pointer to the alpha (slope) tensor.

  • output_dims – [in] Output tensor dimensions.

  • output – [out] Pointer to the output tensor.

Returns:

ARM_CMSIS_NN_SUCCESS on success or ARM_CMSIS_NN_ARG_ERROR on invalid arguments.