Function arm_transpose_conv_f32_get_reverse_conv_buffer_size

• Defined in File arm_nnfunctions_flt.h

Function Documentation

int32_t arm_transpose_conv_f32_get_reverse_conv_buffer_size(const cmsis_nn_transpose_conv_params_f32 *transpose_conv_params, const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims)

Get the reverse-convolution workspace size used by transpose convolution helpers.

Parameters:

• transpose_conv_params – [in] Transpose convolution parameters.

• input_dims – [in] Input tensor dimensions.

• filter_dims – [in] Filter tensor dimensions.

Returns:

Required buffer size in bytes, or 0 when no reverse-convolution buffer is needed.