Function arm_convolve_weight_sum

Function Documentation

arm_cmsis_nn_status arm_convolve_weight_sum(int32_t *vector_sum_buf, const int8_t *rhs, 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-output-channel weight sums for a standard convolution.

  • Supported framework : TensorFlow Lite Micro

  • The buffer pointed to by vector_sum_buf must be at least output_dims->c × sizeof(int32_t) bytes.

Parameters:

  • vector_sum_buf – [out] Pointer to the buffer that will hold the weight sums.

  • rhs – [in] Pointer to the filter weights. Data type: int8

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

  • filter_dims – [in] Filter tensor dimensions. Format: [C_OUT, KH, KW, C_IN]

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

  • lhs_offset – [in] Input-offset added to every input element 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.