Function arm_convolve_1_x_n_nhwc_f16

• Defined in File arm_nnfunctions_flt.h

Function Documentation

arm_cmsis_nn_status arm_convolve_1_x_n_nhwc_f16(const cmsis_nn_context *ctx, const cmsis_nn_conv_params_f16 *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)

1xN convolution, NHWC layout.

Note

When conv_params->weight_format is set to ARM_NN_WEIGHT_FORMAT_NT_N_PACKED, filter_data is interpreted as an already prepacked NTxN RHS buffer instead of the standard public filter layout. Every output position, including the no-pad middle region that OHWI filters feed to the strided direct kernel, is then packed into scratch and multiplied by the format-aware matmul, so a packed 1xN layer with little or no padding runs slower than its OHWI equivalent.