Function arm_transpose_conv_s8_get_reverse_conv_buffer_size¶
Defined in File arm_nnfunctions.h
Function Documentation¶
-
int32_t arm_transpose_conv_s8_get_reverse_conv_buffer_size(const cmsis_nn_transpose_conv_params *transposed_conv_params, const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims)¶
Get the required buffer size for output_ctx in s8 transpose conv function.
- Parameters:
transposed_conv_params – [in] Transposed convolution parameters
input_dims – [in] Input (activation) tensor dimensions. Format: [N, H, W, C_IN]
filter_dims – [in] Filter tensor dimensions. Format: [C_OUT, HK, WK, C_IN] where HK and WK are the spatial filter dimensions
- Returns:
The function returns required buffer size(bytes)