Home
last modified time | relevance | path

Searched refs:in_ch (Results 1 – 3 of 3) sorted by relevance

/hal_nxp-latest/mcux/mcux-sdk/CMSIS/NN/Source/ConvolutionFunctions/
Darm_depthwise_conv_3x3_s8.c99 int32_t in_ch = 0; in arm_depthwise_conv_3x3_s8() local
102 for (; in_ch <= (input_ch - 4); in_ch += 4) in arm_depthwise_conv_3x3_s8()
104 int32_t out_buff0 = bias[in_ch + 0]; in arm_depthwise_conv_3x3_s8()
105 int32_t out_buff1 = bias[in_ch + 1]; in arm_depthwise_conv_3x3_s8()
106 int32_t out_buff2 = bias[in_ch + 2]; in arm_depthwise_conv_3x3_s8()
107 int32_t out_buff3 = bias[in_ch + 3]; in arm_depthwise_conv_3x3_s8()
109 … int8_t *input_ptr = input + (in_h + ker_h_start) * (input_ch * input_x) + in_w * input_ch + in_ch; in arm_depthwise_conv_3x3_s8()
110 const int8_t *kernel_ptr = kernel + ker_h_start * (input_ch * 3) + in_ch; in arm_depthwise_conv_3x3_s8()
151 … out_buff0 = arm_nn_requantize(out_buff0, output_mult[in_ch + 0], output_shift[in_ch + 0]); in arm_depthwise_conv_3x3_s8()
152 … out_buff1 = arm_nn_requantize(out_buff1, output_mult[in_ch + 1], output_shift[in_ch + 1]); in arm_depthwise_conv_3x3_s8()
[all …]
Darm_depthwise_conv_u8_basic_ver1.c72 for (int32_t in_ch = 0, out_ch = 0, ker_w_start = MAX(0, -in_w); out_ch < output_ch; in depthwise_conv_u8_mult_4() local
73 ++in_ch, out_ch += ch_mult) in depthwise_conv_u8_mult_4()
87 … int32_t in_idx = (in_h + ker_h) * (input_ch * input_x) + in_w * input_ch + in_ch; in depthwise_conv_u8_mult_4()
Darm_depthwise_conv_s8.c71 for (int32_t in_ch = 0, out_ch = 0, ker_w_start = MAX(0, -in_w); out_ch < output_ch; in depthwise_conv_s8_mult_4() local
72 ++in_ch, out_ch += ch_mult) in depthwise_conv_s8_mult_4()
86 … int32_t in_idx = (in_h + ker_h) * (input_ch * input_x) + in_w * input_ch + in_ch; in depthwise_conv_s8_mult_4()