arm_transpose_conv_f16.c File Reference

Functions
arm_cmsis_nn_status	arm_transpose_conv_nhwc_f16 (const cmsis_nn_context *ctx, const cmsis_nn_context *output_ctx, const cmsis_nn_transpose_conv_params_f16 *transpose_conv_params, const cmsis_nn_dims *input_dims, const float16_t *input_data, const cmsis_nn_dims *filter_dims, const float16_t *filter_data, const cmsis_nn_dims *bias_dims, const float16_t *bias_data, const cmsis_nn_dims *output_dims, float16_t *output_data)
	Transpose convolution, NHWC layout.
arm_cmsis_nn_status	arm_transpose_conv_f16 (const cmsis_nn_context *ctx, const cmsis_nn_context *output_ctx, const cmsis_nn_transpose_conv_params_f16 *transpose_conv_params, const cmsis_nn_dims *input_dims, const float16_t *input_data, const cmsis_nn_dims *filter_dims, const float16_t *filter_data, const cmsis_nn_dims *bias_dims, const float16_t *bias_data, const cmsis_nn_dims *output_dims, float16_t *output_data, arm_nn_tensor_layout layout)
	Transpose convolution, dispatch by layout.
arm_cmsis_nn_status	arm_transpose_conv_wrapper_f16 (const cmsis_nn_context *ctx, const cmsis_nn_context *output_ctx, const cmsis_nn_transpose_conv_params_f16 *transpose_conv_params, const cmsis_nn_dims *input_dims, const float16_t *input_data, const cmsis_nn_dims *filter_dims, const float16_t *filter_data, const cmsis_nn_dims *bias_dims, const float16_t *bias_data, const cmsis_nn_dims *output_dims, float16_t *output_data, arm_nn_tensor_layout layout)
	Transpose convolution wrapper using the CMSIS-NN baseline path.
int32_t	arm_transpose_conv_f16_get_buffer_size (const cmsis_nn_transpose_conv_params_f16 *transpose_conv_params, const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *out_dims)
	Get the temporary buffer size required by transpose convolution.
int32_t	arm_transpose_conv_f16_get_reverse_conv_buffer_size (const cmsis_nn_transpose_conv_params_f16 *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.