NMSIS-NN  Version 1.2.0
NMSIS NN Software Library
GetBufferSizePooling

Functions

int32_t riscv_avgpool_s16_get_buffer_size (const int output_x, const int ch_src)
 Get the required buffer size for S16 average pooling function. More...
 
int32_t riscv_avgpool_s16_get_buffer_size_dsp (const int output_x, const int ch_src)
 Get the required buffer size for S16 average pooling function for processors with DSP extension. Refer to riscv_avgpool_s16_get_buffer_size() for function argument details. More...
 
int32_t riscv_avgpool_s8_get_buffer_size (const int output_x, const int ch_src)
 Get the required buffer size for S8 average pooling function. More...
 
int32_t riscv_avgpool_s8_get_buffer_size_dsp (const int output_x, const int ch_src)
 Get the required buffer size for S8 average pooling function for processors with DSP extension. Refer to riscv_avgpool_s8_get_buffer_size() for function argument details. More...
 

Detailed Description

Function Documentation

◆ riscv_avgpool_s16_get_buffer_size()

int32_t riscv_avgpool_s16_get_buffer_size ( const int  dim_dst_width,
const int  ch_src 
)

Get the required buffer size for S16 average pooling function.

Parameters
[in]dim_dst_widthoutput tensor dimension
[in]ch_srcnumber of input tensor channels
Returns
The function returns required buffer size in bytes

◆ riscv_avgpool_s16_get_buffer_size_dsp()

int32_t riscv_avgpool_s16_get_buffer_size_dsp ( const int  dim_dst_width,
const int  ch_src 
)

Get the required buffer size for S16 average pooling function for processors with DSP extension. Refer to riscv_avgpool_s16_get_buffer_size() for function argument details.

Note
Intended for compilation on Host. If compiling for an Arm target, use riscv_avgpool_s16_get_buffer_size().

◆ riscv_avgpool_s8_get_buffer_size()

int32_t riscv_avgpool_s8_get_buffer_size ( const int  dim_dst_width,
const int  ch_src 
)

Get the required buffer size for S8 average pooling function.

Parameters
[in]dim_dst_widthoutput tensor dimension
[in]ch_srcnumber of input tensor channels
Returns
The function returns required buffer size in bytes

◆ riscv_avgpool_s8_get_buffer_size_dsp()

int32_t riscv_avgpool_s8_get_buffer_size_dsp ( const int  dim_dst_width,
const int  ch_src 
)

Get the required buffer size for S8 average pooling function for processors with DSP extension. Refer to riscv_avgpool_s8_get_buffer_size() for function argument details.

Note
Intended for compilation on Host. If compiling for an Arm target, use riscv_avgpool_s8_get_buffer_size().