Function arm_convolve_even_s4_get_buffer_size

Function Documentation

int32_t arm_convolve_even_s4_get_buffer_size(const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims)

Get the required buffer size for arm_convolve_even_s4.

Forwards to arm_convolve_s4_get_buffer_size(): the even_s4 kernel stages up to four im2col rows of filter_dims->w * filter_dims->h * input_dims->c int8 elements, byte-for-byte the size that sizer returns. The equality, including the -1 answers for out-of-range shapes, is pinned by a Unity test.

Parameters:

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

  • filter_dims – [in] Filter tensor dimensions. Format: [C_OUT, HK, WK, C_IN] where HK and WK are the spatial filter dimensions

Returns:

The function returns required buffer size in bytes, or -1 if any dimension it reads is negative or the required size would not fit in an int32_t