CMSIS-NN  
CMSIS NN Software Library
 
Loading...
Searching...
No Matches
arm_depthwise_conv_get_buffer_sizes_s4.c File Reference

Functions

int32_t arm_depthwise_conv_s4_opt_get_buffer_size (const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims)
 Get the required buffer size for optimized s4 depthwise convolution function with constraint that in_channel equals out_channel.
 
int32_t arm_depthwise_conv_wrapper_s4_get_buffer_size (const cmsis_nn_dw_conv_params *dw_conv_params, const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims)
 Get size of additional buffer required by arm_depthwise_conv_wrapper_s4()
 
int32_t arm_depthwise_conv_wrapper_s4_get_buffer_size_dsp (const cmsis_nn_dw_conv_params *dw_conv_params, const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims)
 Get size of additional buffer required by arm_depthwise_conv_wrapper_s4() for processors with DSP extension. Refer to arm_depthwise_conv_wrapper_s4_get_buffer_size() for function argument details.
 
int32_t arm_depthwise_conv_wrapper_s4_get_buffer_size_mve (const cmsis_nn_dw_conv_params *dw_conv_params, const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims)
 Get size of additional buffer required by arm_depthwise_conv_wrapper_s4() for Arm(R) Helium Architecture case. Refer to arm_depthwise_conv_wrapper_s4_get_buffer_size() for function argument details.