|
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 *src, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims, int16_t *dst) |
| s16 average pooling function.
|
|
arm_cmsis_nn_status | arm_avgpool_s8 (const cmsis_nn_context *ctx, const cmsis_nn_pool_params *pool_params, const cmsis_nn_dims *input_dims, const int8_t *src, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims, int8_t *dst) |
| s8 average pooling function.
|
|
arm_cmsis_nn_status | arm_max_pool_s16 (const cmsis_nn_context *ctx, const cmsis_nn_pool_params *pool_params, const cmsis_nn_dims *input_dims, const int16_t *src, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims, int16_t *dst) |
| s16 max pooling function.
|
|
arm_cmsis_nn_status | arm_max_pool_s8 (const cmsis_nn_context *ctx, const cmsis_nn_pool_params *pool_params, const cmsis_nn_dims *input_dims, const int8_t *src, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims, int8_t *dst) |
| s8 max pooling function.
|
|
Perform max and average pooling operations
◆ arm_avgpool_s16()
s16 average pooling function.
- Parameters
-
[in,out] | ctx | Function context (e.g. temporary buffer). Check the function definition file to see if an additional buffer is required. Optional function {API}_get_buffer_size() provides the buffer size if an additional buffer is required. The caller is expected to clear the buffer, if applicable, for security reasons. |
[in] | pool_params | Pooling parameters |
[in] | input_dims | Input (activation) tensor dimensions. Format: [H, W, C_IN] |
[in] | input_data | Input (activation) data pointer. Data type: int16 |
[in] | filter_dims | Filter tensor dimensions. Format: [H, W] Argument N and C are not used. |
[in] | output_dims | Output tensor dimensions. Format: [H, W, C_OUT] Argument N is not used. C_OUT equals C_IN. |
[in,out] | output_data | 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
- Supported Framework: TensorFlow Lite
◆ arm_avgpool_s8()
s8 average pooling function.
- Parameters
-
[in,out] | ctx | Function context (e.g. temporary buffer). Check the function definition file to see if an additional buffer is required. Optional function {API}_get_buffer_size() provides the buffer size if an additional buffer is required. The caller is expected to clear the buffer, if applicable, for security reasons. |
[in] | pool_params | Pooling parameters |
[in] | input_dims | Input (activation) tensor dimensions. Format: [H, W, C_IN] |
[in] | input_data | Input (activation) data pointer. Data type: int8 |
[in] | filter_dims | Filter tensor dimensions. Format: [H, W] Argument N and C are not used. |
[in] | output_dims | Output tensor dimensions. Format: [H, W, C_OUT] Argument N is not used. C_OUT equals C_IN. |
[in,out] | output_data | Output data pointer. Data type: int8 |
- Returns
- The function returns either
ARM_CMSIS_NN_ARG_ERROR
if argument constraints fail. or, ARM_CMSIS_NN_SUCCESS
on successful completion.
- Supported Framework: TensorFlow Lite
◆ arm_max_pool_s16()
s16 max pooling function.
- Parameters
-
[in,out] | ctx | Function context (e.g. temporary buffer). Check the function definition file to see if an additional buffer is required. Optional function {API}_get_buffer_size() provides the buffer size if an additional buffer is required. The caller is expected to clear the buffer, if applicable, for security reasons. |
[in] | pool_params | Pooling parameters |
[in] | input_dims | Input (activation) tensor dimensions. Format: [H, W, C_IN] |
[in] | src | Input (activation) data pointer. The input tensor must not overlap with the output tensor. Data type: int16 |
[in] | filter_dims | Filter tensor dimensions. Format: [H, W] Argument N and C are not used. |
[in] | output_dims | Output tensor dimensions. Format: [H, W, C_OUT] Argument N is not used. C_OUT equals C_IN. |
[in,out] | dst | Output data pointer. Data type: int16 |
- Returns
- The function returns either
ARM_CMSIS_NN_ARG_ERROR
if argument constraints fail. or, ARM_CMSIS_NN_SUCCESS
on successful completion.
- Supported Framework: TensorFlow Lite
◆ arm_max_pool_s8()
s8 max pooling function.
- Parameters
-
[in,out] | ctx | Function context (e.g. temporary buffer). Check the function definition file to see if an additional buffer is required. Optional function {API}_get_buffer_size() provides the buffer size if an additional buffer is required. The caller is expected to clear the buffer, if applicable, for security reasons. |
[in] | pool_params | Pooling parameters |
[in] | input_dims | Input (activation) tensor dimensions. Format: [H, W, C_IN] |
[in] | input_data | Input (activation) data pointer. The input tensor must not overlap with the output tensor. Data type: int8 |
[in] | filter_dims | Filter tensor dimensions. Format: [H, W] Argument N and C are not used. |
[in] | output_dims | Output tensor dimensions. Format: [H, W, C_OUT] Argument N is not used. C_OUT equals C_IN. |
[in,out] | output_data | Output data pointer. Data type: int8 |
- Returns
- The function returns either
ARM_CMSIS_NN_ARG_ERROR
if argument constraints fail. or, ARM_CMSIS_NN_SUCCESS
on successful completion.
- Supported Framework: TensorFlow Lite