Searched refs:kernel_size (Results 1 – 7 of 7) sorted by relevance
| /cmsis-nn-latest/Source/ConvolutionFunctions/ |
| D | arm_nn_depthwise_conv_s8_core.c | 48 const uint16_t kernel_size, in arm_nn_depthwise_conv_s8_core() argument
64 int32_t ker_loop = kernel_size / 3; in arm_nn_depthwise_conv_s8_core()
72 const int16_t *col_1 = col + kernel_size * num_ch + offset; in arm_nn_depthwise_conv_s8_core()
114 ker_loop = kernel_size - ((kernel_size / 3) * 3); in arm_nn_depthwise_conv_s8_core()
162 const int32_t single_buffer_size = kernel_size * num_ch; in arm_nn_depthwise_conv_s8_core()
172 for (int j = 0; j < kernel_size; j++) in arm_nn_depthwise_conv_s8_core()
213 (void)kernel_size; in arm_nn_depthwise_conv_s8_core()
|
| D | arm_depthwise_conv_s8_opt.c | 101 const int32_t kernel_size = kernel_x * kernel_y; in arm_depthwise_conv_s8_opt() local
151 kernel_size, in arm_depthwise_conv_s8_opt()
172 … const int8_t *col_0 = lhs_buffer + (kernel_size * CH_IN_BLOCK_MVE * i_buf) + (i_loop_cnt * 4); in arm_depthwise_conv_s8_opt()
180 for (int i_ker = 0; i_ker < kernel_size; i_ker++) in arm_depthwise_conv_s8_opt()
|
| D | arm_depthwise_conv_fast_s16.c | 105 const int32_t kernel_size = kernel_x * kernel_y; in arm_depthwise_conv_fast_s16() local
143 kernel_size, in arm_depthwise_conv_fast_s16()
163 const int16_t *col_0 = lhs_buffer + (kernel_size * input_ch * i_buf) + offset; in arm_depthwise_conv_fast_s16()
167 for (int i_ker = 0; i_ker < kernel_size; i_ker++) in arm_depthwise_conv_fast_s16()
|
| D | arm_depthwise_conv_s4_opt.c | 102 const int32_t kernel_size = kernel_x * kernel_y; in arm_depthwise_conv_s4_opt() local
151 kernel_size, in arm_depthwise_conv_s4_opt()
174 … const int8_t *col_0 = lhs_buffer + (kernel_size * S4_CH_IN_BLOCK_MVE * i_buf) + (i_loop_cnt * 4); in arm_depthwise_conv_s4_opt()
185 for (int i_ker = 0; i_ker < kernel_size; i_ker++) in arm_depthwise_conv_s4_opt()
216 for (int i_ker = 0; i_ker < kernel_size; i_ker++) in arm_depthwise_conv_s4_opt()
|
| /cmsis-nn-latest/Tests/UnitTest/ |
| D | conv_settings.py | 339 kernel_size=(self.filter_y, self.filter_x),
352 … depthwise_layer = keras.layers.DepthwiseConv2D(kernel_size=(self.filter_y, self.filter_x),
366 … kernel_size=(self.filter_y, self.filter_x),
|
| /cmsis-nn-latest/Tests/UnitTest/RefactoredTestGen/Lib/ |
| D | op_conv.py | 79 kernel_size=(params["filter_y"], params["filter_x"]),
|
| /cmsis-nn-latest/Include/ |
| D | arm_nnsupportfunctions.h | 241 const uint16_t kernel_size,
|
