Function arm_convolve_f16_get_buffer_size

• Defined in File arm_nnfunctions_flt.h

Function Documentation

int32_t arm_convolve_f16_get_buffer_size(const cmsis_nn_conv_params_f16 *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 convolution.

Note

When conv_params->weight_format is ARM_NN_WEIGHT_FORMAT_NT_N_PACKED, this still reports only the temporary input/im2col scratch requirement. Any offline-packed filter storage is expected to be provided by the caller.

Parameters:

• conv_params – [in] 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.