Convolution

Conv2D, depthwise, transpose convolution, wrappers, and buffer helpers. 126 functions, declared in 2 modules.

Function	Summary	Module
arm_convolve_1_x_n_f16	1xN convolution, dispatch by layout.	Convolution Functions
arm_convolve_1_x_n_f16_acc16	1xN convolution, dispatch by layout.	Convolution Functions
arm_convolve_1_x_n_f16_get_buffer_size	Get the buffer size required by 1xN convolution.	Convolution Functions
arm_convolve_1_x_n_f32	1xN convolution, dispatch by layout.	Convolution Functions
arm_convolve_1_x_n_f32_get_buffer_size	Get the buffer size required by 1xN convolution.	Convolution Functions
arm_convolve_1_x_n_nhwc_f16	1xN convolution, NHWC layout.	Convolution Functions
arm_convolve_1_x_n_nhwc_f16_acc16	1xN convolution, NHWC layout.	Convolution Functions
arm_convolve_1_x_n_nhwc_f32	1xN convolution, NHWC layout.	Convolution Functions
arm_convolve_1_x_n_s4	1xn convolution for s4 weights	arm_nnfunctions.h
arm_convolve_1_x_n_s4_get_buffer_size	Get the required additional buffer size for 1xn convolution.	arm_nnfunctions.h
arm_convolve_1_x_n_s8	1xn convolution	arm_nnfunctions.h
arm_convolve_1_x_n_s8_get_buffer_size	Get the required additional buffer size for 1xn convolution.	arm_nnfunctions.h
arm_convolve_1x1_f16	1x1 convolution, dispatch by layout.	Convolution Functions
arm_convolve_1x1_f16_acc16	1x1 convolution, dispatch by layout.	Convolution Functions
arm_convolve_1x1_f16_get_buffer_size	Get the buffer size required by 1x1 convolution.	Convolution Functions
arm_convolve_1x1_f32	1x1 convolution, dispatch by layout.	Convolution Functions
arm_convolve_1x1_f32_get_buffer_size	Get the buffer size required by 1x1 convolution.	Convolution Functions
arm_convolve_1x1_nhwc_f16	1x1 convolution, NHWC layout.	Convolution Functions
arm_convolve_1x1_nhwc_f16_acc16	1x1 convolution, NHWC layout.	Convolution Functions
arm_convolve_1x1_nhwc_f32	1x1 convolution, NHWC layout.	Convolution Functions
arm_convolve_1x1_out_s8	Optimised convolution for 1x1 output images (shape of BX1x1xCOUT) for 8x8 computations.	arm_nnfunctions.h
arm_convolve_1x1_out_s8_get_buffer_size	Get the required scratch buffer size for armconvolve1x1outs8().	arm_nnfunctions.h
arm_convolve_1x1_s16_ns_np_nd	Pointwise s16 convolution function: no stride, no padding, no dilation.	arm_nnfunctions.h
arm_convolve_1x1_s4	s4 version for 1x1 convolution with support for non-unity stride values	arm_nnfunctions.h
arm_convolve_1x1_s4_fast	Fast s4 version for 1x1 convolution (non-square shape).	arm_nnfunctions.h
arm_convolve_1x1_s4_fast_get_buffer_size	Get the required buffer size for armconvolve1x1s4fast.	arm_nnfunctions.h
arm_convolve_1x1_s8	s8 version for 1x1 convolution with support for non-unity stride values	arm_nnfunctions.h
arm_convolve_1x1_s8_fast	Fast s8 version for 1x1 convolution (non-square shape).	arm_nnfunctions.h
arm_convolve_1x1_s8_fast_get_buffer_size	Get the required buffer size for armconvolve1x1s8fast.	arm_nnfunctions.h
arm_convolve_even_s4	Basic s4 convolution function with a requirement of even number of kernels.	arm_nnfunctions.h
arm_convolve_even_s4_get_buffer_size	Get the required buffer size for armconvolveevens4.	arm_nnfunctions.h
arm_convolve_f16	Convolution, dispatch by layout.	Convolution Functions
arm_convolve_f16_acc16	Convolution, dispatch by layout.	Convolution Functions
arm_convolve_f16_get_buffer_size	Get the temporary buffer size required by convolution.	Convolution Functions
arm_convolve_f32	Convolution, dispatch by layout.	Convolution Functions
arm_convolve_f32_get_buffer_size	Get the temporary buffer size required by convolution.	Convolution Functions
arm_convolve_nhwc_f16	Convolution, NHWC layout.	Convolution Functions
arm_convolve_nhwc_f16_acc16	Convolution, NHWC layout.	Convolution Functions
arm_convolve_nhwc_f32	Convolution, NHWC layout.	Convolution Functions
arm_convolve_s16	Basic s16 convolution function.	arm_nnfunctions.h
arm_convolve_s16_fast_small_kernel	armconvolves16fastsmallkernel function.	arm_nnfunctions.h
arm_convolve_s16_get_buffer_size	Get the required buffer size for s16 convolution function.	arm_nnfunctions.h
arm_convolve_s16_group_ch_mult_1	s16 grouped convolution optimized for the case where filterdims->c == 1 and inputch == outputch (channel multiplier = 1).	arm_nnfunctions.h
arm_convolve_s4	Basic s4 convolution function.	arm_nnfunctions.h
arm_convolve_s4_get_buffer_size	Get the required buffer size for s4 convolution function.	arm_nnfunctions.h
arm_convolve_s8	Basic s8 convolution function.	arm_nnfunctions.h
arm_convolve_s8_3x3_c16_s1	s8 3x3 convolution over 16 input channels with unit stride.	arm_nnfunctions.h
arm_convolve_s8_get_buffer_size	Get the required buffer size for s8 convolution function.	arm_nnfunctions.h
arm_convolve_s8_get_buffer_size_mve	Get the required buffer size for armconvolves8 for Arm(R) Helium Architecture case.	arm_nnfunctions.h
arm_convolve_s8_get_weights_sum_size	Get the required buffer size for s8 convolution and depthwise convolution weight sum.	arm_nnfunctions.h
arm_convolve_s8_small_cin	s8 convolution for input depths of 1 to 3, such as the first layer of an image or audio model.	arm_nnfunctions.h
arm_convolve_weight_sum	Pre-computes per-output-channel weight sums for a standard convolution.	arm_nnfunctions.h
arm_convolve_wrapper_f16	Convolution wrapper using the CMSIS-NN baseline path.	Convolution Functions
arm_convolve_wrapper_f16_acc16	Convolution wrapper using the CMSIS-NN baseline path.	Convolution Functions
arm_convolve_wrapper_f16_get_buffer_size	Get the buffer size required by the convolution wrapper.	Convolution Functions
arm_convolve_wrapper_f32	Convolution wrapper using the CMSIS-NN baseline path.	Convolution Functions
arm_convolve_wrapper_f32_get_buffer_size	Get the buffer size required by the convolution wrapper.	Convolution Functions
arm_convolve_wrapper_s16	s16 convolution layer wrapper function with the main purpose to call the optimal kernel available in cmsis-nn to perform the convolution.	arm_nnfunctions.h
arm_convolve_wrapper_s16_get_buffer_size	Get the required buffer size for armconvolvewrappers16.	arm_nnfunctions.h
arm_convolve_wrapper_s16_get_buffer_size_dsp	Get the required buffer size for armconvolvewrappers16 for for processors with DSP extension.	arm_nnfunctions.h
arm_convolve_wrapper_s16_get_buffer_size_mve	Get the required buffer size for armconvolvewrappers16 for Arm(R) Helium Architecture case.	arm_nnfunctions.h
arm_convolve_wrapper_s4	s4 convolution layer wrapper function with the main purpose to call the optimal kernel available in cmsis-nn to perform the convolution.	arm_nnfunctions.h
arm_convolve_wrapper_s4_get_buffer_size	Get the required buffer size for armconvolvewrappers4.	arm_nnfunctions.h
arm_convolve_wrapper_s4_get_buffer_size_dsp	Get the required buffer size for armconvolvewrappers4 for processors with DSP extension.	arm_nnfunctions.h
arm_convolve_wrapper_s4_get_buffer_size_mve	Get the required buffer size for armconvolvewrappers4 for Arm(R) Helium Architecture case.	arm_nnfunctions.h
arm_convolve_wrapper_s8	s8 convolution layer wrapper function with the main purpose to call the optimal kernel available in cmsis-nn to perform the convolution.	arm_nnfunctions.h
arm_convolve_wrapper_s8_get_buffer_size	Get the required buffer size for armconvolvewrappers8.	arm_nnfunctions.h
arm_convolve_wrapper_s8_get_buffer_size_dsp	Get the required buffer size for armconvolvewrappers8 for processors with DSP extension.	arm_nnfunctions.h
arm_convolve_wrapper_s8_get_buffer_size_mve	Get the required buffer size for armconvolvewrappers8 for Arm(R) Helium Architecture case.	arm_nnfunctions.h
arm_depthwise_conv_3x3_s8	Optimized s8 depthwise convolution function for 3x3 kernel size with some constraints on the input arguments(documented below).	arm_nnfunctions.h
arm_depthwise_conv_f16	Depthwise convolution, dispatch by layout.	Convolution Functions
arm_depthwise_conv_f16_acc16	Depthwise convolution, dispatch by layout.	Convolution Functions
arm_depthwise_conv_f16_get_buffer_size	Get the temporary buffer size required by depthwise convolution.	Convolution Functions
arm_depthwise_conv_f32	Depthwise convolution, dispatch by layout.	Convolution Functions
arm_depthwise_conv_f32_get_buffer_size	Get the temporary buffer size required by depthwise convolution.	Convolution Functions
arm_depthwise_conv_fast_s16	Optimized s16 depthwise convolution function with constraint that inchannel equals outchannel.	arm_nnfunctions.h
arm_depthwise_conv_fast_s16_get_buffer_size	Get the required buffer size for optimized s16 depthwise convolution function with constraint that inchannel equals outchannel.	arm_nnfunctions.h
arm_depthwise_conv_s16	Basic s16 depthwise convolution function that doesn’t have any constraints on the input dimensions.	arm_nnfunctions.h
arm_depthwise_conv_s4	Basic s4 depthwise convolution function that doesn’t have any constraints on the input dimensions.	arm_nnfunctions.h
arm_depthwise_conv_s4_opt	Optimized s4 depthwise convolution function with constraint that inchannel equals outchannel.	arm_nnfunctions.h
arm_depthwise_conv_s4_opt_get_buffer_size	Get the required buffer size for optimized s4 depthwise convolution function with constraint that inchannel equals outchannel.	arm_nnfunctions.h
arm_depthwise_conv_s8	Basic s8 depthwise convolution function that doesn’t have any constraints on the input dimensions.	arm_nnfunctions.h
arm_depthwise_conv_s8_opt	Optimized s8 depthwise convolution function with constraint that inchannel equals outchannel.	arm_nnfunctions.h
arm_depthwise_conv_s8_opt_3x3	s8 3x3 depthwise convolution that reads the input in place, without the im2col copy of armdepthwiseconvs8opt(), for layers in its gate.	arm_nnfunctions.h
arm_depthwise_conv_s8_opt_3x3_c64_s1	armdepthwiseconvs8opt3x3() specialized for 64 channels and a vertical stride of 1, with fixed channel offsets and edge columns that skip their zero taps.	arm_nnfunctions.h
arm_depthwise_conv_s8_opt_3x3_get_buffer_size	Get the scratch size in bytes of armdepthwiseconvs8opt3x3() and armdepthwiseconvs8opt3x3c64s1().	arm_nnfunctions.h
arm_depthwise_conv_s8_opt_channelwise	The channel-vectorized path of armdepthwiseconvs8opt() on its own, without the planar attempt.	arm_nnfunctions.h
arm_depthwise_conv_s8_opt_get_buffer_size	Get the required buffer size for optimized s8 depthwise convolution function with constraint that inchannel equals outchannel.	arm_nnfunctions.h
arm_depthwise_conv_s8_opt_planar	The planar path of armdepthwiseconvs8opt() on its own: s8 depthwise convolution vectorized across the output pixels of each channel plane.	arm_nnfunctions.h
arm_depthwise_conv_s8_opt_planar_supported	Whether armdepthwiseconvs8opt() runs a layer on its planar path, vectorized across the output pixels of each channel plane rather than across channels.	arm_nnfunctions.h
arm_depthwise_conv_wrapper_f16	Depthwise convolution wrapper using the CMSIS-NN baseline path.	Convolution Functions
arm_depthwise_conv_wrapper_f16_acc16	Depthwise convolution wrapper using the CMSIS-NN baseline path.	Convolution Functions
arm_depthwise_conv_wrapper_f16_get_buffer_size	Get the buffer size required by the depthwise convolution wrapper.	Convolution Functions
arm_depthwise_conv_wrapper_f32	Depthwise convolution wrapper using the CMSIS-NN baseline path.	Convolution Functions
arm_depthwise_conv_wrapper_f32_get_buffer_size	Get the buffer size required by the depthwise convolution wrapper.	Convolution Functions
arm_depthwise_conv_wrapper_s16	Wrapper function to pick the right optimized s16 depthwise convolution function.	arm_nnfunctions.h
arm_depthwise_conv_wrapper_s16_get_buffer_size	Get size of additional buffer required by armdepthwiseconvwrappers16().	arm_nnfunctions.h
arm_depthwise_conv_wrapper_s16_get_buffer_size_dsp	Get size of additional buffer required by armdepthwiseconvwrappers16() for processors with DSP extension.	arm_nnfunctions.h
arm_depthwise_conv_wrapper_s16_get_buffer_size_mve	Get size of additional buffer required by armdepthwiseconvwrappers16() for Arm(R) Helium Architecture case.	arm_nnfunctions.h
arm_depthwise_conv_wrapper_s4	Wrapper function to pick the right optimized s4 depthwise convolution function.	arm_nnfunctions.h
arm_depthwise_conv_wrapper_s4_get_buffer_size	Get size of additional buffer required by armdepthwiseconvwrappers4().	arm_nnfunctions.h
arm_depthwise_conv_wrapper_s4_get_buffer_size_dsp	Get size of additional buffer required by armdepthwiseconvwrappers4() for processors with DSP extension.	arm_nnfunctions.h
arm_depthwise_conv_wrapper_s4_get_buffer_size_mve	Get size of additional buffer required by armdepthwiseconvwrappers4() for Arm(R) Helium Architecture case.	arm_nnfunctions.h
arm_depthwise_conv_wrapper_s8	Wrapper function to pick the right optimized s8 depthwise convolution function.	arm_nnfunctions.h
arm_depthwise_conv_wrapper_s8_get_buffer_size	Get size of additional buffer required by armdepthwiseconvwrappers8().	arm_nnfunctions.h
arm_depthwise_conv_wrapper_s8_get_buffer_size_dsp	Get size of additional buffer required by armdepthwiseconvwrappers8() for processors with DSP extension.	arm_nnfunctions.h
arm_depthwise_conv_wrapper_s8_get_buffer_size_mve	Get size of additional buffer required by armdepthwiseconvwrappers8() for Arm(R) Helium Architecture case.	arm_nnfunctions.h
arm_depthwise_convolve_weight_sum	Pre-computes per-channel weight sums for a depthwise convolution.	arm_nnfunctions.h
arm_depthwise_nhwc_conv_f16	Depthwise convolution, NHWC layout.	Convolution Functions
arm_depthwise_nhwc_conv_f16_acc16	Depthwise convolution, NHWC layout.	Convolution Functions
arm_depthwise_nhwc_conv_f32	Depthwise convolution, NHWC layout.	Convolution Functions
arm_transpose_conv_f16	Transpose convolution, dispatch by layout.	Convolution Functions
arm_transpose_conv_f16_get_buffer_size	Get the temporary buffer size required by transpose convolution.	Convolution Functions
arm_transpose_conv_f16_get_reverse_conv_buffer_size	Get the reverse-convolution workspace size used by transpose convolution helpers.	Convolution Functions
arm_transpose_conv_f32	Transpose convolution, dispatch by layout.	Convolution Functions
arm_transpose_conv_f32_get_buffer_size	Get the temporary buffer size required by transpose convolution.	Convolution Functions
arm_transpose_conv_f32_get_reverse_conv_buffer_size	Get the reverse-convolution workspace size used by transpose convolution helpers.	Convolution Functions
arm_transpose_conv_nhwc_f16	Transpose convolution, NHWC layout.	Convolution Functions
arm_transpose_conv_nhwc_f32	Transpose convolution, NHWC layout.	Convolution Functions
arm_transpose_conv_s8	Basic s8 transpose convolution function.	arm_nnfunctions.h
arm_transpose_conv_s8_get_buffer_size	Get the required buffer size for ctx in s8 transpose conv function.	arm_nnfunctions.h
arm_transpose_conv_s8_get_buffer_size_mve	Get size of additional buffer required by armtransposeconvs8() for Arm(R) Helium Architecture case.	arm_nnfunctions.h
arm_transpose_conv_s8_get_reverse_conv_buffer_size	Get the required buffer size for outputctx in s8 transpose conv function.	arm_nnfunctions.h
arm_transpose_conv_wrapper_f16	Transpose convolution wrapper using the CMSIS-NN baseline path.	Convolution Functions
arm_transpose_conv_wrapper_f32	Transpose convolution wrapper using the CMSIS-NN baseline path.	Convolution Functions
arm_transpose_conv_wrapper_s8	Wrapper to select optimal transposed convolution algorithm depending on parameters.	arm_nnfunctions.h

All heliaCORE functions in one page