CMSIS-NN  
CMSIS NN Software Library
arm_depthwise_conv_wrapper_s16.c File Reference

Macros

#define USE_FAST_DW_CONV_FUNCTION(dw_conv_params, filter_dims, input_dims)
 

Functions

arm_cmsis_nn_status arm_depthwise_conv_wrapper_s16 (const cmsis_nn_context *ctx, const cmsis_nn_dw_conv_params *dw_conv_params, const cmsis_nn_per_channel_quant_params *quant_params, const cmsis_nn_dims *input_dims, const int16_t *input, const cmsis_nn_dims *filter_dims, const int8_t *filter, const cmsis_nn_dims *bias_dims, const int64_t *bias, const cmsis_nn_dims *output_dims, int16_t *output)
 Wrapper function to pick the right optimized s16 depthwise convolution function. More...
 
int32_t arm_depthwise_conv_wrapper_s16_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_s16() More...
 

Macro Definition Documentation

◆ USE_FAST_DW_CONV_FUNCTION

#define USE_FAST_DW_CONV_FUNCTION (   dw_conv_params,
  filter_dims,
  input_dims 
)
Value:
(dw_conv_params->ch_mult == 1 && dw_conv_params->dilation.w == 1 && dw_conv_params->dilation.h == 1 && \
filter_dims->w * filter_dims->h * input_dims->c < 512)