Searched refs:kernel_size (Results 1 – 5 of 5) sorted by relevance
48 const uint16_t kernel_size, in arm_nn_depthwise_conv_s8_core() argument64 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()
101 const int32_t kernel_size = kernel_x * kernel_y; in arm_depthwise_conv_s8_opt() local151 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()
105 const int32_t kernel_size = kernel_x * kernel_y; in arm_depthwise_conv_fast_s16() local143 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()
192 const uint16_t kernel_size,
702 kernel_size=(self.filter_y, self.filter_x),710 … depthwise_layer = tf.keras.layers.DepthwiseConv2D(kernel_size=(self.filter_y, self.filter_x),
