arm_depthwise_conv_f16.c File Reference

Macros
#define	ARM_NN_DW_NT_T_F16_TILE_ROWS   (4)

Functions
arm_cmsis_nn_status	arm_depthwise_nhwc_conv_f16 (const cmsis_nn_context *ctx, const cmsis_nn_dw_conv_params_f16 *dw_conv_params, const cmsis_nn_dims *input_dims, const float16_t *input, const cmsis_nn_dims *filter_dims, const float16_t *kernel, const cmsis_nn_dims *bias_dims, const float16_t *bias, const cmsis_nn_dims *output_dims, float16_t *output)
	Depthwise convolution, NHWC layout.
arm_cmsis_nn_status	arm_depthwise_conv_f16 (const cmsis_nn_context *ctx, const cmsis_nn_dw_conv_params_f16 *dw_conv_params, const cmsis_nn_dims *input_dims, const float16_t *input, const cmsis_nn_dims *filter_dims, const float16_t *kernel, const cmsis_nn_dims *bias_dims, const float16_t *bias, const cmsis_nn_dims *output_dims, float16_t *output, arm_nn_tensor_layout layout)
	Depthwise convolution, dispatch by layout.
arm_cmsis_nn_status	arm_depthwise_conv_wrapper_f16 (const cmsis_nn_context *ctx, const cmsis_nn_dw_conv_params_f16 *dw_conv_params, const cmsis_nn_dims *input_dims, const float16_t *input, const cmsis_nn_dims *filter_dims, const float16_t *kernel, const cmsis_nn_dims *bias_dims, const float16_t *bias, const cmsis_nn_dims *output_dims, float16_t *output)
	Depthwise convolution wrapper using the CMSIS-NN baseline path.