Function arm_avgpool_s16_get_buffer_size¶
Defined in File arm_nnfunctions.h
Function Documentation¶
-
int32_t arm_avgpool_s16_get_buffer_size(const int dim_dst_width, const int ch_src)¶
Get the required buffer size for S16 average pooling function.
As in the s8 variant, it is the DSP leg that carries a byte count here: for a valid (non-negative, in-range) ch_src this returns ch_src * sizeof(int32_t) on builds with the DSP extension but no MVE, and 0 on builds with MVE and on plain-C builds. For an invalid ch_src it returns -1 on every build target.
- Parameters:
dim_dst_width – [in] output tensor dimension
ch_src – [in] number of input tensor channels
- Returns:
The function returns required buffer size in bytes, or -1 if ch_src is negative or the required size would not fit in an int32_t