Function arm_depthwise_conv_wrapper_s16_get_buffer_size

• Defined in File arm_nnfunctions.h

Function Documentation

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

Parameters:

• dw_conv_params – [in] Depthwise convolution parameters (e.g. strides, dilations, pads,…) Range of dw_conv_params->input_offset : Not used Range of dw_conv_params->input_offset : Not used

• input_dims – [in] Input (activation) tensor dimensions. Format: [H, W, C_IN] Batch argument N is not used and assumed to be 1.

• filter_dims – [in] Filter tensor dimensions. Format: [1, H, W, C_OUT]

• output_dims – [in] Output tensor dimensions. Format: [1, H, W, C_OUT]

Returns:

Size of additional memory required for optimizations in bytes.