Function arm_fully_connected_f32_get_buffer_size¶

Defined in File arm_nnfunctions_flt.h

Function Documentation¶

int32_t arm_fully_connected_f32_get_buffer_size(const cmsis_nn_fc_params_f32 *fc_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 the fully connected layer.

Parameters:

fc_params – [in] Fully connected 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.