Function arm_avgpool_s16

Function Documentation

arm_cmsis_nn_status arm_avgpool_s16(const cmsis_nn_context *ctx, const cmsis_nn_pool_params *pool_params, const cmsis_nn_dims *input_dims, const int16_t *input_data, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims, int16_t *output_data)

s16 average pooling function.

  • Supported Framework: TensorFlow Lite

Parameters:

  • ctx – [inout] Function context. Size ctx->buf with arm_avgpool_s16_get_buffer_size(output_dims->w, input_dims->c). Note that it takes the output width and the input channel count rather than a cmsis_nn_dims. arm_avgpool_s16_get_buffer_size_dsp() and arm_avgpool_s16_get_buffer_size_mve() size the same buffer for a specific target. It returns 0 where no buffer is needed, in which case ctx->buf may be NULL. The caller is expected to clear the buffer, if applicable, for security reasons.

  • pool_params – [in] Pooling parameters

  • input_dims – [in] Input (activation) tensor dimensions. Format: [H, W, C_IN]

  • input_data – [in] Input (activation) data pointer. Data type: int16

  • filter_dims – [in] Filter tensor dimensions. Format: [H, W] Argument N and C are not used.

  • output_dims – [in] Output tensor dimensions. Format: [H, W, C_OUT] Argument N is not used. C_OUT equals C_IN.

  • output_data – [inout] Output data pointer. Data type: int16

Returns:

The function returns ARM_CMSIS_NN_SUCCESS - Successful operation ARM_CMSIS_NN_ARG_ERROR - In case of invalid arguments