Function arm_convolve_s8_get_weights_sum_size

Function Documentation

int32_t arm_convolve_s8_get_weights_sum_size(const cmsis_nn_dims *output_dims)

Get the required buffer size for s8 convolution and depthwise convolution weight sum.

For a valid (non-negative, in-range) output_dims->c, returns output_dims->c * sizeof(int32_t) on builds with the MVE extension and 0 elsewhere. A negative or out-of-range output_dims->c returns -1 on builds with the MVE extension; elsewhere no weight sum buffer is used and the answer stays 0.

Parameters:

  • output_dims – [in] Output (activation) tensor dimensions. Format: [N, H, W, C_COUT]

Returns:

The function returns required weight sum buffer size in bytes, or -1 if output_dims->c is negative or the required size would not fit in an int32_t