Function arm_depthwise_convolve_weight_sum¶

Function Documentation¶

arm_cmsis_nn_status arm_depthwise_convolve_weight_sum(int32_t *vector_sum_buf, int8_t *scratch_buf, const int8_t *rhs, const cmsis_nn_dw_conv_params *dw_conv_params, const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims, const int32_t lhs_offset, const int32_t *bias_data)¶

Pre-computes per-channel weight sums for a depthwise convolution.

Parameters:

vector_sum_buf – [out] Buffer to hold the computed weight sums.
scratch_buf – [inout] Scratch buffer used internally; caller must clear for security.
rhs – [in] Depthwise convolution weights. Data type: int8
dw_conv_params – [in] Depthwise-convolution parameters (stride, dilation, pad, etc.)
input_dims – [in] Input tensor dimensions. Format: [N, H, W, C_IN]
filter_dims – [in] Filter tensor dimensions. Format: [1, KH, KW, C_OUT]
output_dims – [in] Output tensor dimensions. Format: [N, H, W, C_OUT]
lhs_offset – [in] Input-offset applied before MAC. Range: [-127, 128]
bias_data – [in] Optional bias pointer. Data type: int32

Returns:

ARM_CMSIS_NN_ARG_ERROR on invalid arguments, or ARM_CMSIS_NN_SUCCESS on success.