arm_get_buffer_size_f16.c File Reference

Macros
#define	ARM_NN_DW_NT_T_F16_TILE_ROWS   (4)

Functions
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.
int32_t	arm_depthwise_conv_wrapper_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)
	Get the buffer size required by the depthwise convolution wrapper.
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.
int32_t	arm_convolve_wrapper_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)
	Get the buffer size required by the convolution wrapper.
int32_t	arm_convolve_1x1_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 buffer size required by 1x1 convolution.
int32_t	arm_convolve_1_x_n_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 buffer size required by 1xN convolution.
int32_t	arm_fully_connected_f16_get_buffer_size (const cmsis_nn_fc_params_f16 *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.
int32_t	arm_batch_matmul_f16_get_buffer_size (const cmsis_nn_bmm_params_f16 *bmm_params, const cmsis_nn_dims *input_lhs_dims, const cmsis_nn_dims *input_rhs_dims, const cmsis_nn_dims *output_dims)
	Get the temporary buffer size required by batched matrix multiplication.