Function arm_depthwise_conv_f16_get_buffer_size¶

Defined in File arm_nnfunctions_flt.h

Function Documentation¶

int32_t arm_depthwise_conv_f16_get_buffer_size(const cmsis_nn_dw_conv_params_f16 *dw_conv_params, const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims, arm_nn_tensor_layout layout)¶

Get the temporary buffer size required by depthwise convolution.

Parameters:

dw_conv_params – [in] Depthwise convolution parameters.
input_dims – [in] Input tensor dimensions.
filter_dims – [in] Filter tensor dimensions.
output_dims – [in] Output tensor dimensions.
layout – [in] Tensor layout selector.

Returns:

Required buffer size in bytes, or 0 when no scratch buffer is needed.