Home
last modified time | relevance | path

Searched refs:ch_src (Results 1 – 5 of 5) sorted by relevance

/cmsis-nn-latest/Source/PoolingFunctions/
Darm_avgpool_get_buffer_sizes_s16.c42 int32_t arm_avgpool_s16_get_buffer_size(const int output_x, const int ch_src) in arm_avgpool_s16_get_buffer_size() argument
45 return arm_avgpool_s16_get_buffer_size_mve(output_x, ch_src); in arm_avgpool_s16_get_buffer_size()
47 return arm_avgpool_s16_get_buffer_size_dsp(output_x, ch_src); in arm_avgpool_s16_get_buffer_size()
50 (void)ch_src; in arm_avgpool_s16_get_buffer_size()
55 int32_t arm_avgpool_s16_get_buffer_size_dsp(const int output_x, const int ch_src) in arm_avgpool_s16_get_buffer_size_dsp() argument
58 return (ch_src * sizeof(int32_t)); in arm_avgpool_s16_get_buffer_size_dsp()
61 int32_t arm_avgpool_s16_get_buffer_size_mve(const int output_x, const int ch_src) in arm_avgpool_s16_get_buffer_size_mve() argument
64 (void)ch_src; in arm_avgpool_s16_get_buffer_size_mve()
Darm_avgpool_get_buffer_sizes_s8.c42 int32_t arm_avgpool_s8_get_buffer_size(const int output_x, const int ch_src) in arm_avgpool_s8_get_buffer_size() argument
45 return arm_avgpool_s8_get_buffer_size_mve(output_x, ch_src); in arm_avgpool_s8_get_buffer_size()
47 return arm_avgpool_s8_get_buffer_size_dsp(output_x, ch_src); in arm_avgpool_s8_get_buffer_size()
50 (void)ch_src; in arm_avgpool_s8_get_buffer_size()
55 int32_t arm_avgpool_s8_get_buffer_size_dsp(const int output_x, const int ch_src) in arm_avgpool_s8_get_buffer_size_dsp() argument
58 return (ch_src * sizeof(int32_t)); in arm_avgpool_s8_get_buffer_size_dsp()
61 int32_t arm_avgpool_s8_get_buffer_size_mve(const int output_x, const int ch_src) in arm_avgpool_s8_get_buffer_size_mve() argument
64 (void)ch_src; in arm_avgpool_s8_get_buffer_size_mve()
Darm_avgpool_s16.c93 const int32_t ch_src = input_dims->c; in arm_avgpool_s16() local
94 const int32_t batch_input = input_x * input_y * ch_src; in arm_avgpool_s16()
105 const int32_t batch_output = output_x * output_y * ch_src; in arm_avgpool_s16()
120 int16_t *out = &dst[ch_src * (i_x + i_y * output_x)]; in arm_avgpool_s16()
122 int32_t ch_count = (ch_src + 7) / 8; in arm_avgpool_s16()
123 int32_t channels = ch_src; in arm_avgpool_s16()
139 const int16_t *src_inner = src_base + (ch_src * (k_x + k_y * input_x)); in arm_avgpool_s16()
233 … const int16_t *start = src + ch_src * (k_x + idx_x + (k_y + idx_y) * input_x); in arm_avgpool_s16()
237 for (int i = 0; i < ch_src; i++) in arm_avgpool_s16()
244 for (int i = 0; i < ch_src; i++) in arm_avgpool_s16()
[all …]
Darm_avgpool_s8.c95 const int32_t ch_src = input_dims->c; in arm_avgpool_s8() local
96 const int32_t batch_input = input_x * input_y * ch_src; in arm_avgpool_s8()
97 const int32_t batch_output = output_x * output_y * ch_src; in arm_avgpool_s8()
118 int8_t *out = &dst[ch_src * (i_x + i_y * output_x)]; in arm_avgpool_s8()
120 int32_t ch_count = (ch_src + 15) / 16; in arm_avgpool_s8()
121 int32_t channels = ch_src; in arm_avgpool_s8()
142 const int8_t *src_inner = src_base + (ch_src * (k_x + k_y * input_x)); in arm_avgpool_s8()
256 const int32_t ch_src = input_dims->c; in arm_avgpool_s8() local
272 const int32_t batch_size = input_x * input_y * ch_src; in arm_avgpool_s8()
297 … const int8_t *start = src + ch_src * (k_x + idx_x + (k_y + idx_y) * input_x); in arm_avgpool_s8()
[all …]
/cmsis-nn-latest/Include/
Darm_nnfunctions.h1921 int32_t arm_avgpool_s8_get_buffer_size(const int dim_dst_width, const int ch_src);
1931 int32_t arm_avgpool_s8_get_buffer_size_dsp(const int dim_dst_width, const int ch_src);
1941 int32_t arm_avgpool_s8_get_buffer_size_mve(const int dim_dst_width, const int ch_src);
1984 int32_t arm_avgpool_s16_get_buffer_size(const int dim_dst_width, const int ch_src);
1994 int32_t arm_avgpool_s16_get_buffer_size_dsp(const int dim_dst_width, const int ch_src);
2004 int32_t arm_avgpool_s16_get_buffer_size_mve(const int dim_dst_width, const int ch_src);