Function arm_convolve_s16_group_ch_mult_1

Function Documentation

arm_cmsis_nn_status arm_convolve_s16_group_ch_mult_1(const cmsis_nn_context *ctx, const cmsis_nn_conv_params *conv_params, const cmsis_nn_per_channel_quant_params *quant_params, const cmsis_nn_dims *input_dims, const int16_t *input_data, const cmsis_nn_dims *filter_dims, const int8_t *filter_data, const cmsis_nn_dims *bias_dims, const cmsis_nn_bias_data *bias_data, const cmsis_nn_dims *output_dims, int16_t *output_data)

s16 grouped convolution optimized for the case where filter_dims->c == 1 and input_ch == output_ch (channel multiplier = 1).

Parameters:

  • ctx – [inout] Function context (unused, pass NULL-initialised).

  • conv_params – [in] Convolution parameters (strides, dilations, pads, activation).

  • quant_params – [in] Per-channel quantization info (multiplier and shift).

  • input_dims – [in] Input tensor dimensions. Format: [N, H, W, C_IN]

  • input_data – [in] Input data pointer. Data type: int16

  • filter_dims – [in] Filter tensor dimensions. Format: [C_OUT, HK, WK, 1]

  • filter_data – [in] Filter data pointer. Data type: int8

  • bias_dims – [in] Bias tensor dimensions (unused, may be zero-initialised).

  • bias_data – [in] Optional bias struct (int32 or int64). May be NULL.

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

  • output_data – [out] Output data pointer. Data type: int16

Returns:

ARM_CMSIS_NN_SUCCESS on success.