| /hal_nxp-3.6.0/mcux/mcux-sdk/CMSIS/NN/Source/ConvolutionFunctions/ |
| D | arm_convolve_wrapper_s8.c | 53 const cmsis_nn_dims *input_dims, in arm_convolve_wrapper_s8() argument
62 … if ((conv_params->padding.w == 0) && (conv_params->padding.h == 0) && (input_dims->c % 4 == 0) && in arm_convolve_wrapper_s8()
68 input_dims, in arm_convolve_wrapper_s8()
77 …else if ((output_dims->h == 1) && (input_dims->h == 1) && (filter_dims->h == 1) && (output_dims->w… in arm_convolve_wrapper_s8()
78 (input_dims->n == 1)) in arm_convolve_wrapper_s8()
83 input_dims, in arm_convolve_wrapper_s8()
97 input_dims, in arm_convolve_wrapper_s8()
109 const cmsis_nn_dims *input_dims, in arm_convolve_wrapper_s8_get_buffer_size() argument
113 … if ((conv_params->padding.w == 0) && (conv_params->padding.h == 0) && (input_dims->c % 4 == 0) && in arm_convolve_wrapper_s8_get_buffer_size()
116 return arm_convolve_1x1_s8_fast_get_buffer_size(input_dims); in arm_convolve_wrapper_s8_get_buffer_size()
[all …]
|
| D | arm_convolve_1x1_s8_fast.c | 56 const cmsis_nn_dims *input_dims, in arm_convolve_1x1_s8_fast() argument
65 if (input_dims->c % 4 != 0 || conv_params->padding.w != 0 || conv_params->padding.h != 0 || in arm_convolve_1x1_s8_fast()
77 const int32_t col_len = input_dims->w * input_dims->h * input_dims->n; in arm_convolve_1x1_s8_fast()
79 const int32_t input_ch = input_dims->c; in arm_convolve_1x1_s8_fast()
149 const int32_t lhs_rows = input_dims->w * input_dims->h * input_dims->n; in arm_convolve_1x1_s8_fast()
151 const int32_t rhs_cols = input_dims->c; in arm_convolve_1x1_s8_fast()
173 int32_t arm_convolve_1x1_s8_fast_get_buffer_size(const cmsis_nn_dims *input_dims) in arm_convolve_1x1_s8_fast_get_buffer_size() argument
175 (void)input_dims; in arm_convolve_1x1_s8_fast_get_buffer_size()
|
| D | arm_depthwise_conv_wrapper_s8.c | 52 const cmsis_nn_dims *input_dims, in arm_depthwise_conv_wrapper_s8() argument
62 if (1 == dw_conv_params->ch_mult && input_dims->n == 1) in arm_depthwise_conv_wrapper_s8()
70 input_dims, in arm_depthwise_conv_wrapper_s8()
85 input_dims, in arm_depthwise_conv_wrapper_s8()
100 input_dims, in arm_depthwise_conv_wrapper_s8()
115 const cmsis_nn_dims *input_dims, in arm_depthwise_conv_wrapper_s8_get_buffer_size() argument
122 if (input_dims->c == output_dims->c && input_dims->n == 1) in arm_depthwise_conv_wrapper_s8_get_buffer_size()
124 size = arm_depthwise_conv_s8_opt_get_buffer_size(input_dims, filter_dims); in arm_depthwise_conv_wrapper_s8_get_buffer_size()
|
| D | arm_convolve_1_x_n_s8.c | 53 const cmsis_nn_dims *input_dims, in arm_convolve_1_x_n_s8() argument
73 const uint16_t input_x = input_dims->w; in arm_convolve_1_x_n_s8()
77 const uint16_t input_ch = input_dims->c; in arm_convolve_1_x_n_s8()
175 input_dims, in arm_convolve_1_x_n_s8()
190 int32_t arm_convolve_1_x_n_s8_get_buffer_size(const cmsis_nn_dims *input_dims, const cmsis_nn_dims … in arm_convolve_1_x_n_s8_get_buffer_size() argument
193 return (2 * input_dims->c * filter_dims->w * filter_dims->h) * sizeof(int16_t); in arm_convolve_1_x_n_s8_get_buffer_size()
195 (void)input_dims; in arm_convolve_1_x_n_s8_get_buffer_size()
|
| D | arm_depthwise_conv_s8.c | 232 const cmsis_nn_dims *input_dims, in arm_depthwise_conv_s8() argument
245 if (dw_conv_params->ch_mult % 4 == 0 && input_dims->n == 1) in arm_depthwise_conv_s8()
248 input_dims->w, in arm_depthwise_conv_s8()
249 input_dims->h, in arm_depthwise_conv_s8()
250 input_dims->c, in arm_depthwise_conv_s8()
274 input_dims->n, in arm_depthwise_conv_s8()
275 input_dims->w, in arm_depthwise_conv_s8()
276 input_dims->h, in arm_depthwise_conv_s8()
277 input_dims->c, in arm_depthwise_conv_s8()
|
| D | arm_depthwise_conv_s8_opt.c | 54 const cmsis_nn_dims *input_dims, in arm_depthwise_conv_s8_opt() argument
64 const int32_t input_ch = input_dims->c; in arm_depthwise_conv_s8_opt()
73 const int32_t input_x = input_dims->w; in arm_depthwise_conv_s8_opt()
74 const int32_t input_y = input_dims->h; in arm_depthwise_conv_s8_opt()
398 input_dims, in arm_depthwise_conv_s8_opt()
412 int32_t arm_depthwise_conv_s8_opt_get_buffer_size(const cmsis_nn_dims *input_dims, const cmsis_nn_d… in arm_depthwise_conv_s8_opt_get_buffer_size() argument
416 return (2 * input_dims->c * filter_dims->w * filter_dims->h) * (int32_t)sizeof(int16_t) + 4; in arm_depthwise_conv_s8_opt_get_buffer_size()
418 return (input_dims->c * filter_dims->w * filter_dims->h) * sizeof(int16_t); in arm_depthwise_conv_s8_opt_get_buffer_size()
420 (void)input_dims; in arm_depthwise_conv_s8_opt_get_buffer_size()
|
| D | arm_convolve_s8.c | 54 const cmsis_nn_dims *input_dims, in arm_convolve_s8() argument
66 const uint16_t input_batches = input_dims->n; in arm_convolve_s8()
67 const uint16_t input_x = input_dims->w; in arm_convolve_s8()
68 const uint16_t input_y = input_dims->h; in arm_convolve_s8()
69 const uint16_t input_ch = input_dims->c; in arm_convolve_s8()
367 int32_t arm_convolve_s8_get_buffer_size(const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filte… in arm_convolve_s8_get_buffer_size() argument
370 return (2 * input_dims->c * filter_dims->w * filter_dims->h) * (int32_t)sizeof(int16_t); in arm_convolve_s8_get_buffer_size()
372 (void)input_dims; in arm_convolve_s8_get_buffer_size()
|
| D | arm_depthwise_conv_3x3_s8.c | 55 const cmsis_nn_dims *input_dims, in arm_depthwise_conv_3x3_s8() argument
67 const int32_t input_x = input_dims->w; in arm_depthwise_conv_3x3_s8()
68 const int32_t input_y = input_dims->h; in arm_depthwise_conv_3x3_s8()
69 const int32_t input_ch = input_dims->c; in arm_depthwise_conv_3x3_s8()
|
| /hal_nxp-3.6.0/mcux/mcux-sdk/CMSIS/NN/Include/ |
| D | arm_nnfunctions.h | 197 const cmsis_nn_dims *input_dims,
221 const cmsis_nn_dims *input_dims,
255 const cmsis_nn_dims *input_dims,
273 int32_t arm_convolve_s8_get_buffer_size(const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filte…
575 const cmsis_nn_dims *input_dims,
591 int32_t arm_convolve_1x1_s8_fast_get_buffer_size(const cmsis_nn_dims *input_dims);
632 const cmsis_nn_dims *input_dims,
650 int32_t arm_convolve_1_x_n_s8_get_buffer_size(const cmsis_nn_dims *input_dims, const cmsis_nn_dims …
929 const cmsis_nn_dims *input_dims,
953 const cmsis_nn_dims *input_dims,
[all …]
|
| /hal_nxp-3.6.0/mcux/mcux-sdk/CMSIS/NN/Source/PoolingFunctions/ |
| D | arm_avgpool_s8.c | 76 const cmsis_nn_dims *input_dims, in arm_avgpool_s8() argument
83 const int32_t input_y = input_dims->h; in arm_avgpool_s8()
84 const int32_t input_x = input_dims->w; in arm_avgpool_s8()
95 const int32_t ch_src = input_dims->c; in arm_avgpool_s8()
245 const cmsis_nn_dims *input_dims, in arm_avgpool_s8() argument
251 const int32_t input_y = input_dims->h; in arm_avgpool_s8()
252 const int32_t input_x = input_dims->w; in arm_avgpool_s8()
263 const int32_t ch_src = input_dims->c; in arm_avgpool_s8()
|
| D | arm_max_pool_s8.c | 164 const cmsis_nn_dims *input_dims, in arm_max_pool_s8() argument
170 const int32_t input_y = input_dims->h; in arm_max_pool_s8()
171 const int32_t input_x = input_dims->w; in arm_max_pool_s8()
182 const int32_t channel_in = input_dims->c; in arm_max_pool_s8()
|
| /hal_nxp-3.6.0/mcux/mcux-sdk/CMSIS/NN/Source/FullyConnectedFunctions/ |
| D | arm_fully_connected_s8.c | 53 const cmsis_nn_dims *input_dims, in arm_fully_connected_s8() argument
66 int32_t batch_cnt = input_dims->n; in arm_fully_connected_s8()
|
| /hal_nxp-3.6.0/mcux/mcux-sdk/CMSIS/NN/Source/SVDFunctions/ |
| D | arm_svdf_s8.c | 55 const cmsis_nn_dims *input_dims, in arm_svdf_s8() argument
84 const int32_t input_batches = input_dims->n; in arm_svdf_s8()
85 const int32_t input_height = input_dims->h; in arm_svdf_s8()
|
