Home
last modified time | relevance | path

Searched refs:output_dims (Results 1 – 13 of 13) sorted by relevance

/hal_nxp-latest/mcux/mcux-sdk/CMSIS/NN/Source/ConvolutionFunctions/
Darm_convolve_wrapper_s8.c59 const cmsis_nn_dims *output_dims, in arm_convolve_wrapper_s8() argument
74 output_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()
89 output_dims, in arm_convolve_wrapper_s8()
103 output_dims, in arm_convolve_wrapper_s8()
111 const cmsis_nn_dims *output_dims) in arm_convolve_wrapper_s8_get_buffer_size() argument
118 …else if ((output_dims->h == 1) && (input_dims->h == 1) && (filter_dims->h == 1) && (output_dims->w… in arm_convolve_wrapper_s8_get_buffer_size()
Darm_depthwise_conv_wrapper_s8.c58 const cmsis_nn_dims *output_dims, in arm_depthwise_conv_wrapper_s8() argument
76 output_dims, in arm_depthwise_conv_wrapper_s8()
91 output_dims, in arm_depthwise_conv_wrapper_s8()
106 output_dims, in arm_depthwise_conv_wrapper_s8()
117 const cmsis_nn_dims *output_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()
Darm_depthwise_conv_s8.c238 const cmsis_nn_dims *output_dims, in arm_depthwise_conv_s8() argument
252 output_dims->c, in arm_depthwise_conv_s8()
264 output_dims->w, in arm_depthwise_conv_s8()
265 output_dims->h, in arm_depthwise_conv_s8()
279 output_dims->c, in arm_depthwise_conv_s8()
291 output_dims->w, in arm_depthwise_conv_s8()
292 output_dims->h, in arm_depthwise_conv_s8()
Darm_convolve_1_x_n_s8.c59 const cmsis_nn_dims *output_dims, in arm_convolve_1_x_n_s8() argument
64 if (output_dims->w % 4 != 0) in arm_convolve_1_x_n_s8()
75 const uint16_t output_x = output_dims->w; in arm_convolve_1_x_n_s8()
76 const uint16_t output_ch = output_dims->c; in arm_convolve_1_x_n_s8()
181 output_dims, in arm_convolve_1_x_n_s8()
Darm_depthwise_conv_3x3_s8.c61 const cmsis_nn_dims *output_dims, in arm_depthwise_conv_3x3_s8() argument
70 const int32_t output_ch = output_dims->c; in arm_depthwise_conv_3x3_s8()
77 const int32_t output_x = output_dims->w; in arm_depthwise_conv_3x3_s8()
78 const int32_t output_y = output_dims->h; in arm_depthwise_conv_3x3_s8()
Darm_depthwise_conv_s8_opt.c60 const cmsis_nn_dims *output_dims, in arm_depthwise_conv_s8_opt() argument
65 const int32_t output_ch = output_dims->c; in arm_depthwise_conv_s8_opt()
83 const int32_t output_x = output_dims->w; in arm_depthwise_conv_s8_opt()
84 const int32_t output_y = output_dims->h; in arm_depthwise_conv_s8_opt()
404 output_dims, in arm_depthwise_conv_s8_opt()
Darm_convolve_1x1_s8_fast.c62 const cmsis_nn_dims *output_dims, in arm_convolve_1x1_s8_fast() argument
78 const int32_t output_ch = output_dims->c; in arm_convolve_1x1_s8_fast()
150 const int32_t rhs_rows = output_dims->c; in arm_convolve_1x1_s8_fast()
Darm_convolve_s8.c60 const cmsis_nn_dims *output_dims, in arm_convolve_s8() argument
72 const uint16_t output_x = output_dims->w; in arm_convolve_s8()
73 const uint16_t output_y = output_dims->h; in arm_convolve_s8()
74 const uint16_t output_ch = output_dims->c; in arm_convolve_s8()
/hal_nxp-latest/mcux/mcux-sdk/CMSIS/NN/Include/
Darm_nnfunctions.h203 const cmsis_nn_dims *output_dims,
223 const cmsis_nn_dims *output_dims);
261 const cmsis_nn_dims *output_dims,
581 const cmsis_nn_dims *output_dims,
638 const cmsis_nn_dims *output_dims,
935 const cmsis_nn_dims *output_dims,
955 const cmsis_nn_dims *output_dims);
996 const cmsis_nn_dims *output_dims,
1026 const cmsis_nn_dims *output_dims,
1061 const cmsis_nn_dims *output_dims,
[all …]
/hal_nxp-latest/mcux/mcux-sdk/CMSIS/NN/Source/PoolingFunctions/
Darm_avgpool_s8.c79 const cmsis_nn_dims *output_dims, in arm_avgpool_s8() argument
85 const int32_t output_y = output_dims->h; in arm_avgpool_s8()
86 const int32_t output_x = output_dims->w; in arm_avgpool_s8()
248 const cmsis_nn_dims *output_dims, in arm_avgpool_s8() argument
253 const int32_t output_y = output_dims->h; in arm_avgpool_s8()
254 const int32_t output_x = output_dims->w; in arm_avgpool_s8()
Darm_max_pool_s8.c167 const cmsis_nn_dims *output_dims, in arm_max_pool_s8() argument
172 const int32_t output_y = output_dims->h; in arm_max_pool_s8()
173 const int32_t output_x = output_dims->w; in arm_max_pool_s8()
/hal_nxp-latest/mcux/mcux-sdk/CMSIS/NN/Source/FullyConnectedFunctions/
Darm_fully_connected_s8.c59 const cmsis_nn_dims *output_dims, in arm_fully_connected_s8() argument
80 output_dims->c, /* row_dim or output_depth */ in arm_fully_connected_s8()
84 output += output_dims->c; in arm_fully_connected_s8()
/hal_nxp-latest/mcux/mcux-sdk/CMSIS/NN/Source/SVDFunctions/
Darm_svdf_s8.c65 const cmsis_nn_dims *output_dims, in arm_svdf_s8() argument
70 (void)output_dims; in arm_svdf_s8()