Function arm_convolve_s4_get_buffer_size¶

Defined in File arm_nnfunctions.h

Function Documentation¶

int32_t arm_convolve_s4_get_buffer_size(const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims)¶

Get the required buffer size for s4 convolution function.

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(bytes)