CMSIS-NN
CMSIS NN Software Library
|
Functions | |
int32_t | arm_convolve_fast_s16_get_buffer_size (const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims) |
Get the required buffer size for fast s16 convolution function. More... | |
int32_t | arm_convolve_s16_get_buffer_size (const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims) |
Get the required buffer size for s16 convolution function. More... | |
int32_t | arm_convolve_wrapper_s16_get_buffer_size (const cmsis_nn_conv_params *conv_params, const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims) |
Get the required buffer size for arm_convolve_wrapper_s16. More... | |
int32_t | arm_convolve_wrapper_s16_get_buffer_size_dsp (const cmsis_nn_conv_params *conv_params, const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims) |
Get the required buffer size for arm_convolve_wrapper_s16 for for processors with DSP extension. Refer to arm_convolve_wrapper_s16_get_buffer_size() for function argument details. More... | |
int32_t | arm_convolve_wrapper_s16_get_buffer_size_mve (const cmsis_nn_conv_params *conv_params, const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims) |
Get the required buffer size for arm_convolve_wrapper_s16 for Arm(R) Helium Architecture case. Refer to arm_convolve_wrapper_s16_get_buffer_size() for function argument details. More... | |
int32_t | arm_convolve_s4_get_buffer_size (const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims) |
Get the required buffer size for s4 convolution function. More... | |
int32_t | arm_convolve_1x1_s4_fast_get_buffer_size (const cmsis_nn_dims *input_dims) |
Get the required buffer size for arm_convolve_1x1_s4_fast. More... | |
int32_t | arm_convolve_wrapper_s4_get_buffer_size (const cmsis_nn_conv_params *conv_params, const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims) |
Get the required buffer size for arm_convolve_wrapper_s4. More... | |
int32_t | arm_convolve_wrapper_s4_get_buffer_size_mve (const cmsis_nn_conv_params *conv_params, const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims) |
Get the required buffer size for arm_convolve_wrapper_s4 for Arm(R) Helium Architecture case. Refer to arm_convolve_wrapper_s4_get_buffer_size() for function argument details. More... | |
int32_t | arm_convolve_wrapper_s4_get_buffer_size_dsp (const cmsis_nn_conv_params *conv_params, const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims) |
Get the required buffer size for arm_convolve_wrapper_s4 for processors with DSP extension. Refer to arm_convolve_wrapper_s4_get_buffer_size() for function argument details. More... | |
int32_t | arm_convolve_s8_get_buffer_size (const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims) |
Get the required buffer size for s8 convolution function. More... | |
int32_t | arm_convolve_1_x_n_s8_get_buffer_size (const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims) |
Get the required additional buffer size for 1xn convolution. More... | |
int32_t | arm_convolve_1x1_s8_fast_get_buffer_size (const cmsis_nn_dims *input_dims) |
Get the required buffer size for arm_convolve_1x1_s8_fast. More... | |
int32_t | arm_convolve_wrapper_s8_get_buffer_size (const cmsis_nn_conv_params *conv_params, const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims) |
Get the required buffer size for arm_convolve_wrapper_s8. More... | |
int32_t | arm_convolve_wrapper_s8_get_buffer_size_mve (const cmsis_nn_conv_params *conv_params, const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims) |
Get the required buffer size for arm_convolve_wrapper_s8 for Arm(R) Helium Architecture case. Refer to arm_convolve_wrapper_s8_get_buffer_size() for function argument details. More... | |
int32_t | arm_convolve_wrapper_s8_get_buffer_size_dsp (const cmsis_nn_conv_params *conv_params, const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims) |
Get the required buffer size for arm_convolve_wrapper_s8 for processors with DSP extension. Refer to arm_convolve_wrapper_s8_get_buffer_size() for function argument details. More... | |
int32_t | arm_depthwise_conv_fast_s16_get_buffer_size (const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims) |
Get the required buffer size for optimized s16 depthwise convolution function with constraint that in_channel equals out_channel. More... | |
int32_t | arm_depthwise_conv_wrapper_s16_get_buffer_size (const cmsis_nn_dw_conv_params *dw_conv_params, const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims) |
Get size of additional buffer required by arm_depthwise_conv_wrapper_s16() More... | |
int32_t | arm_depthwise_conv_wrapper_s16_get_buffer_size_mve (const cmsis_nn_dw_conv_params *dw_conv_params, const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims) |
Get size of additional buffer required by arm_depthwise_conv_wrapper_s16() for Arm(R) Helium Architecture case. Refer to arm_depthwise_conv_wrapper_s16_get_buffer_size() for function argument details. More... | |
int32_t | arm_depthwise_conv_wrapper_s16_get_buffer_size_dsp (const cmsis_nn_dw_conv_params *dw_conv_params, const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims) |
Get size of additional buffer required by arm_depthwise_conv_wrapper_s16() for processors with DSP extension. Refer to arm_depthwise_conv_wrapper_s16_get_buffer_size() for function argument details. More... | |
int32_t | arm_depthwise_conv_s4_opt_get_buffer_size (const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims) |
Get the required buffer size for optimized s4 depthwise convolution function with constraint that in_channel equals out_channel. More... | |
int32_t | arm_depthwise_conv_wrapper_s4_get_buffer_size (const cmsis_nn_dw_conv_params *dw_conv_params, const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims) |
Get size of additional buffer required by arm_depthwise_conv_wrapper_s4() More... | |
int32_t | arm_depthwise_conv_wrapper_s4_get_buffer_size_dsp (const cmsis_nn_dw_conv_params *dw_conv_params, const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims) |
Get size of additional buffer required by arm_depthwise_conv_wrapper_s4() for processors with DSP extension. Refer to arm_depthwise_conv_wrapper_s4_get_buffer_size() for function argument details. More... | |
int32_t | arm_depthwise_conv_wrapper_s4_get_buffer_size_mve (const cmsis_nn_dw_conv_params *dw_conv_params, const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims) |
Get size of additional buffer required by arm_depthwise_conv_wrapper_s4() for Arm(R) Helium Architecture case. Refer to arm_depthwise_conv_wrapper_s4_get_buffer_size() for function argument details. More... | |
int32_t | arm_depthwise_conv_s8_opt_get_buffer_size_dsp (const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims) |
Get the required buffer size for optimized s8 depthwise convolution function with constraint that in_channel equals out_channel. This is for processors with DSP extension. Refer to arm_depthwise_conv_s8_opt_get_buffer_size() for function argument details. More... | |
int32_t | arm_depthwise_conv_s8_opt_get_buffer_size (const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims) |
Get the required buffer size for optimized s8 depthwise convolution function with constraint that in_channel equals out_channel. More... | |
int32_t | arm_depthwise_conv_wrapper_s8_get_buffer_size (const cmsis_nn_dw_conv_params *dw_conv_params, const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims) |
Get size of additional buffer required by arm_depthwise_conv_wrapper_s8() More... | |
int32_t | arm_depthwise_conv_wrapper_s8_get_buffer_size_dsp (const cmsis_nn_dw_conv_params *dw_conv_params, const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims) |
Get size of additional buffer required by arm_depthwise_conv_wrapper_s8() for processors with DSP extension. Refer to arm_depthwise_conv_wrapper_s8_get_buffer_size() for function argument details. More... | |
int32_t | arm_depthwise_conv_wrapper_s8_get_buffer_size_mve (const cmsis_nn_dw_conv_params *dw_conv_params, const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims) |
Get size of additional buffer required by arm_depthwise_conv_wrapper_s8() for Arm(R) Helium Architecture case. Refer to arm_depthwise_conv_wrapper_s8_get_buffer_size() for function argument details. More... | |
int32_t | arm_transpose_conv_s8_get_buffer_size (const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *out_dims) |
Get the required buffer size for s8 transpose conv function. More... | |
int32_t | arm_transpose_conv_s8_get_buffer_size_dsp (const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims) |
Get size of additional buffer required by arm_transpose_conv_s8() for processors with DSP extension. Refer to arm_transpose_conv_s8_get_buffer_size() for function argument details. More... | |
int32_t | arm_transpose_conv_s8_get_buffer_size_mve (const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filter_dims, const cmsis_nn_dims *output_dims) |
Get size of additional buffer required by arm_transpose_conv_s8() for Arm(R) Helium Architecture case. Refer to arm_transpose_conv_s8_get_buffer_size() for function argument details. More... | |
int32_t arm_convolve_1_x_n_s8_get_buffer_size | ( | const cmsis_nn_dims * | input_dims, |
const cmsis_nn_dims * | filter_dims | ||
) |
[in] | input_dims | Input (activation) tensor dimensions. Format: [N, H, W, C_IN] |
[in] | filter_dims | Filter tensor dimensions. Format: [C_OUT, 1, WK, C_IN] where WK is the horizontal spatial filter dimension |
int32_t arm_convolve_1x1_s4_fast_get_buffer_size | ( | const cmsis_nn_dims * | input_dims | ) |
[in] | input_dims | Input (activation) dimensions |
int32_t arm_convolve_1x1_s8_fast_get_buffer_size | ( | const cmsis_nn_dims * | input_dims | ) |
[in] | input_dims | Input (activation) dimensions |
int32_t arm_convolve_fast_s16_get_buffer_size | ( | const cmsis_nn_dims * | input_dims, |
const cmsis_nn_dims * | filter_dims | ||
) |
[in] | input_dims | Input (activation) tensor dimensions. Format: [N, H, W, C_IN] |
[in] | filter_dims | Filter tensor dimensions. Format: [C_OUT, HK, WK, C_IN] where HK and WK are the spatial filter dimensions |
int32_t arm_convolve_s16_get_buffer_size | ( | const cmsis_nn_dims * | input_dims, |
const cmsis_nn_dims * | filter_dims | ||
) |
[in] | input_dims | Input (activation) tensor dimensions. Format: [N, H, W, C_IN] |
[in] | filter_dims | Filter tensor dimensions. Format: [C_OUT, HK, WK, C_IN] where HK and WK are the spatial filter dimensions |
int32_t arm_convolve_s4_get_buffer_size | ( | const cmsis_nn_dims * | input_dims, |
const cmsis_nn_dims * | filter_dims | ||
) |
[in] | input_dims | Input (activation) tensor dimensions. Format: [N, H, W, C_IN] |
[in] | filter_dims | Filter tensor dimensions. Format: [C_OUT, HK, WK, C_IN] where HK and WK are the spatial filter dimensions |
int32_t arm_convolve_s8_get_buffer_size | ( | const cmsis_nn_dims * | input_dims, |
const cmsis_nn_dims * | filter_dims | ||
) |
[in] | input_dims | Input (activation) tensor dimensions. Format: [N, H, W, C_IN] |
[in] | filter_dims | Filter tensor dimensions. Format: [C_OUT, HK, WK, C_IN] where HK and WK are the spatial filter dimensions |
int32_t arm_convolve_wrapper_s16_get_buffer_size | ( | const cmsis_nn_conv_params * | conv_params, |
const cmsis_nn_dims * | input_dims, | ||
const cmsis_nn_dims * | filter_dims, | ||
const cmsis_nn_dims * | output_dims | ||
) |
[in] | conv_params | Convolution parameters (e.g. strides, dilations, pads,...). conv_params->input_offset : Not used conv_params->output_offset : Not used |
[in] | input_dims | Input (activation) dimensions. Format: [N, H, W, C_IN] |
[in] | filter_dims | Filter dimensions. Format: [C_OUT, HK, WK, C_IN] where HK and WK are the spatial filter dimensions |
[in] | output_dims | Output tensor dimensions. Format: [N, H, W, C_OUT] |
int32_t arm_convolve_wrapper_s16_get_buffer_size_dsp | ( | const cmsis_nn_conv_params * | conv_params, |
const cmsis_nn_dims * | input_dims, | ||
const cmsis_nn_dims * | filter_dims, | ||
const cmsis_nn_dims * | output_dims | ||
) |
int32_t arm_convolve_wrapper_s16_get_buffer_size_mve | ( | const cmsis_nn_conv_params * | conv_params, |
const cmsis_nn_dims * | input_dims, | ||
const cmsis_nn_dims * | filter_dims, | ||
const cmsis_nn_dims * | output_dims | ||
) |
int32_t arm_convolve_wrapper_s4_get_buffer_size | ( | const cmsis_nn_conv_params * | conv_params, |
const cmsis_nn_dims * | input_dims, | ||
const cmsis_nn_dims * | filter_dims, | ||
const cmsis_nn_dims * | output_dims | ||
) |
[in] | conv_params | Convolution parameters (e.g. strides, dilations, pads,...). Range of conv_params->input_offset : [-127, 128] Range of conv_params->output_offset : [-128, 127] |
[in] | input_dims | Input (activation) dimensions. Format: [N, H, W, C_IN] |
[in] | filter_dims | Filter dimensions. Format: [C_OUT, HK, WK, C_IN] where HK and WK are the spatial filter dimensions |
[in] | output_dims | Output tensor dimensions. Format: [N, H, W, C_OUT] |
int32_t arm_convolve_wrapper_s4_get_buffer_size_dsp | ( | const cmsis_nn_conv_params * | conv_params, |
const cmsis_nn_dims * | input_dims, | ||
const cmsis_nn_dims * | filter_dims, | ||
const cmsis_nn_dims * | output_dims | ||
) |
int32_t arm_convolve_wrapper_s4_get_buffer_size_mve | ( | const cmsis_nn_conv_params * | conv_params, |
const cmsis_nn_dims * | input_dims, | ||
const cmsis_nn_dims * | filter_dims, | ||
const cmsis_nn_dims * | output_dims | ||
) |
int32_t arm_convolve_wrapper_s8_get_buffer_size | ( | const cmsis_nn_conv_params * | conv_params, |
const cmsis_nn_dims * | input_dims, | ||
const cmsis_nn_dims * | filter_dims, | ||
const cmsis_nn_dims * | output_dims | ||
) |
[in] | conv_params | Convolution parameters (e.g. strides, dilations, pads,...). Range of conv_params->input_offset : [-127, 128] Range of conv_params->output_offset : [-128, 127] |
[in] | input_dims | Input (activation) dimensions. Format: [N, H, W, C_IN] |
[in] | filter_dims | Filter dimensions. Format: [C_OUT, HK, WK, C_IN] where HK and WK are the spatial filter dimensions |
[in] | output_dims | Output tensor dimensions. Format: [N, H, W, C_OUT] |
int32_t arm_convolve_wrapper_s8_get_buffer_size_dsp | ( | const cmsis_nn_conv_params * | conv_params, |
const cmsis_nn_dims * | input_dims, | ||
const cmsis_nn_dims * | filter_dims, | ||
const cmsis_nn_dims * | output_dims | ||
) |
int32_t arm_convolve_wrapper_s8_get_buffer_size_mve | ( | const cmsis_nn_conv_params * | conv_params, |
const cmsis_nn_dims * | input_dims, | ||
const cmsis_nn_dims * | filter_dims, | ||
const cmsis_nn_dims * | output_dims | ||
) |
int32_t arm_depthwise_conv_fast_s16_get_buffer_size | ( | const cmsis_nn_dims * | input_dims, |
const cmsis_nn_dims * | filter_dims | ||
) |
[in] | input_dims | Input (activation) tensor dimensions. Format: [1, H, W, C_IN] Batch argument N is not used. |
[in] | filter_dims | Filter tensor dimensions. Format: [1, H, W, C_OUT] |
int32_t arm_depthwise_conv_s4_opt_get_buffer_size | ( | const cmsis_nn_dims * | input_dims, |
const cmsis_nn_dims * | filter_dims | ||
) |
[in] | input_dims | Input (activation) tensor dimensions. Format: [1, H, W, C_IN] Batch argument N is not used. |
[in] | filter_dims | Filter tensor dimensions. Format: [1, H, W, C_OUT] |
int32_t arm_depthwise_conv_s8_opt_get_buffer_size | ( | const cmsis_nn_dims * | input_dims, |
const cmsis_nn_dims * | filter_dims | ||
) |
[in] | input_dims | Input (activation) tensor dimensions. Format: [1, H, W, C_IN] Batch argument N is not used. |
[in] | filter_dims | Filter tensor dimensions. Format: [1, H, W, C_OUT] |
int32_t arm_depthwise_conv_s8_opt_get_buffer_size_dsp | ( | const cmsis_nn_dims * | input_dims, |
const cmsis_nn_dims * | filter_dims | ||
) |
int32_t arm_depthwise_conv_wrapper_s16_get_buffer_size | ( | const cmsis_nn_dw_conv_params * | dw_conv_params, |
const cmsis_nn_dims * | input_dims, | ||
const cmsis_nn_dims * | filter_dims, | ||
const cmsis_nn_dims * | output_dims | ||
) |
[in] | dw_conv_params | Depthwise convolution parameters (e.g. strides, dilations, pads,...) Range of dw_conv_params->input_offset : Not used Range of dw_conv_params->input_offset : Not used |
[in] | input_dims | Input (activation) tensor dimensions. Format: [H, W, C_IN] Batch argument N is not used and assumed to be 1. |
[in] | filter_dims | Filter tensor dimensions. Format: [1, H, W, C_OUT] |
[in] | output_dims | Output tensor dimensions. Format: [1, H, W, C_OUT] |
int32_t arm_depthwise_conv_wrapper_s16_get_buffer_size_dsp | ( | const cmsis_nn_dw_conv_params * | dw_conv_params, |
const cmsis_nn_dims * | input_dims, | ||
const cmsis_nn_dims * | filter_dims, | ||
const cmsis_nn_dims * | output_dims | ||
) |
int32_t arm_depthwise_conv_wrapper_s16_get_buffer_size_mve | ( | const cmsis_nn_dw_conv_params * | dw_conv_params, |
const cmsis_nn_dims * | input_dims, | ||
const cmsis_nn_dims * | filter_dims, | ||
const cmsis_nn_dims * | output_dims | ||
) |
int32_t arm_depthwise_conv_wrapper_s4_get_buffer_size | ( | const cmsis_nn_dw_conv_params * | dw_conv_params, |
const cmsis_nn_dims * | input_dims, | ||
const cmsis_nn_dims * | filter_dims, | ||
const cmsis_nn_dims * | output_dims | ||
) |
[in] | dw_conv_params | Depthwise convolution parameters (e.g. strides, dilations, pads,...) Range of dw_conv_params->input_offset : [-127, 128] Range of dw_conv_params->input_offset : [-128, 127] |
[in] | input_dims | Input (activation) tensor dimensions. Format: [H, W, C_IN] Batch argument N is not used and assumed to be 1. |
[in] | filter_dims | Filter tensor dimensions. Format: [1, H, W, C_OUT] |
[in] | output_dims | Output tensor dimensions. Format: [1, H, W, C_OUT] |
int32_t arm_depthwise_conv_wrapper_s4_get_buffer_size_dsp | ( | const cmsis_nn_dw_conv_params * | dw_conv_params, |
const cmsis_nn_dims * | input_dims, | ||
const cmsis_nn_dims * | filter_dims, | ||
const cmsis_nn_dims * | output_dims | ||
) |
int32_t arm_depthwise_conv_wrapper_s4_get_buffer_size_mve | ( | const cmsis_nn_dw_conv_params * | dw_conv_params, |
const cmsis_nn_dims * | input_dims, | ||
const cmsis_nn_dims * | filter_dims, | ||
const cmsis_nn_dims * | output_dims | ||
) |
int32_t arm_depthwise_conv_wrapper_s8_get_buffer_size | ( | const cmsis_nn_dw_conv_params * | dw_conv_params, |
const cmsis_nn_dims * | input_dims, | ||
const cmsis_nn_dims * | filter_dims, | ||
const cmsis_nn_dims * | output_dims | ||
) |
[in] | dw_conv_params | Depthwise convolution parameters (e.g. strides, dilations, pads,...) Range of dw_conv_params->input_offset : [-127, 128] Range of dw_conv_params->input_offset : [-128, 127] |
[in] | input_dims | Input (activation) tensor dimensions. Format: [H, W, C_IN] Batch argument N is not used and assumed to be 1. |
[in] | filter_dims | Filter tensor dimensions. Format: [1, H, W, C_OUT] |
[in] | output_dims | Output tensor dimensions. Format: [1, H, W, C_OUT] |
int32_t arm_depthwise_conv_wrapper_s8_get_buffer_size_dsp | ( | const cmsis_nn_dw_conv_params * | dw_conv_params, |
const cmsis_nn_dims * | input_dims, | ||
const cmsis_nn_dims * | filter_dims, | ||
const cmsis_nn_dims * | output_dims | ||
) |
int32_t arm_depthwise_conv_wrapper_s8_get_buffer_size_mve | ( | const cmsis_nn_dw_conv_params * | dw_conv_params, |
const cmsis_nn_dims * | input_dims, | ||
const cmsis_nn_dims * | filter_dims, | ||
const cmsis_nn_dims * | output_dims | ||
) |
int32_t arm_transpose_conv_s8_get_buffer_size | ( | const cmsis_nn_dims * | input_dims, |
const cmsis_nn_dims * | filter_dims, | ||
const cmsis_nn_dims * | out_dims | ||
) |
[in] | input_dims | Input (activation) tensor dimensions. Format: [N, H, W, C_IN] |
[in] | filter_dims | Filter tensor dimensions. Format: [C_OUT, HK, WK, C_IN] where HK and WK are the spatial filter dimensions |
[in] | out_dims | Output tensor dimensions. Format: [N, H, W, C_OUT] |
int32_t arm_transpose_conv_s8_get_buffer_size_dsp | ( | const cmsis_nn_dims * | input_dims, |
const cmsis_nn_dims * | filter_dims, | ||
const cmsis_nn_dims * | out_dims | ||
) |
int32_t arm_transpose_conv_s8_get_buffer_size_mve | ( | const cmsis_nn_dims * | input_dims, |
const cmsis_nn_dims * | filter_dims, | ||
const cmsis_nn_dims * | out_dims | ||
) |