Function arm_convolve_wrapper_s4_get_buffer_size¶
Defined in File arm_nnfunctions.h
Function Documentation¶
-
int32_t arm_convolve_wrapper_s4_get_buffer_size(const cmsis_nn_conv_params *conv_params, const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims)¶
Get the required buffer size for arm_convolve_wrapper_s4.
- Parameters:
conv_params – [in] Convolution parameters (e.g. strides, dilations, pads,…). Range of conv_params->input_offset : [-127, 128] Range of conv_params->output_offset : [-128, 127]
input_dims – [in] Input (activation) dimensions. Format: [N, H, W, C_IN]
filter_dims – [in] Filter dimensions. Format: [C_OUT, HK, WK, C_IN] where HK and WK are the spatial filter dimensions
output_dims – [in] Output tensor dimensions. Format: [N, H, W, C_OUT]
- Returns:
The function returns required buffer size(bytes)