| /cmsis-nn-latest/Tests/UnitTest/RefactoredTestGen/JsonTemplates/ |
| D | fully_connected.json | 35 out_ch,
56 out_ch
76 out_ch
|
| D | fully_connected_null_bias.json | 35 out_ch,
57 out_ch
|
| /cmsis-nn-latest/Source/NNSupportFunctions/ |
| D | arm_nn_mat_mul_core_4x_s8.c | 51 const int32_t out_ch, in arm_nn_mat_mul_core_4x_s8() argument
59 for (int i = 0; i < out_ch; i++) in arm_nn_mat_mul_core_4x_s8()
128 …const uint32x4_t scatter_offset = {0, (uint32_t)out_ch, (uint32_t)out_ch * 2, (uint32_t)out_ch * 3… in arm_nn_mat_mul_core_4x_s8()
133 return output + (3 * out_ch); in arm_nn_mat_mul_core_4x_s8()
139 (void)out_ch; in arm_nn_mat_mul_core_4x_s8()
|
| D | arm_nn_mat_mul_core_1x_s8.c | 51 const int32_t out_ch, in arm_nn_mat_mul_core_1x_s8() argument
66 for (int i = 0; i < out_ch; i++) in arm_nn_mat_mul_core_1x_s8()
122 for (int i = 0; i < (out_ch & 0x3); i++) in arm_nn_mat_mul_core_1x_s8()
142 (void)out_ch; in arm_nn_mat_mul_core_1x_s8()
|
| D | arm_nn_mat_mul_core_1x_s4.c | 51 const int32_t out_ch, in arm_nn_mat_mul_core_1x_s4() argument
69 for (int i = 0; i < out_ch; i++) in arm_nn_mat_mul_core_1x_s4()
118 for (int i = 0; i < (out_ch & 0x3); i++) in arm_nn_mat_mul_core_1x_s4()
138 (void)out_ch; in arm_nn_mat_mul_core_1x_s4()
|
| /cmsis-nn-latest/Tests/UnitTest/ |
| D | generate_test_data.py | 84 out_ch=1,
101 out_ch=1,
118 out_ch=7,
140 out_ch=5,
160 out_ch=15,
181 out_ch=5,
201 out_ch=5,
221 out_ch=1,
240 out_ch=3,
260 out_ch=3,
[all …]
|
| D | fully_connected_settings.py | 34 out_ch=1, argument
69 out_ch,
107 "bias_size": out_ch,
110 "output_size": out_ch,
|
| D | conv_settings.py | 35 out_ch=1, argument
68 out_ch,
107 … raise RuntimeError("out channel ({}) is not multiple of in channel ({})".format(out_ch, in_ch))
116 if out_ch % groups != 0:
117 …timeError("ERROR: Output channels {} must be an even multiple of groups {}".format(out_ch, groups))
|
| D | add_mul_settings.py | 50 out_ch=channels,
|
| D | test_settings.py | 66 out_ch, argument
147 self.output_ch = out_ch
|
| /cmsis-nn-latest/Source/ConvolutionFunctions/ |
| D | arm_depthwise_conv_s16.c | 69 for (int32_t in_ch = 0, out_ch = 0, ker_w_start = MAX(0, -in_w); out_ch < output_ch; in depthwise_conv_s16_mult_4_s16() local
70 ++in_ch, out_ch += ch_mult) in depthwise_conv_s16_mult_4_s16()
74 int32_t out_buff32[4] = {REDUCE_MULTIPLIER(output_mult[out_ch + 0 + mult_tile]), in depthwise_conv_s16_mult_4_s16()
75 REDUCE_MULTIPLIER(output_mult[out_ch + 1 + mult_tile]), in depthwise_conv_s16_mult_4_s16()
76 REDUCE_MULTIPLIER(output_mult[out_ch + 2 + mult_tile]), in depthwise_conv_s16_mult_4_s16()
77 … REDUCE_MULTIPLIER(output_mult[out_ch + 3 + mult_tile])}; in depthwise_conv_s16_mult_4_s16()
83 out_buff[0] = bias[out_ch + 0 + mult_tile]; in depthwise_conv_s16_mult_4_s16()
84 out_buff[1] = bias[out_ch + 1 + mult_tile]; in depthwise_conv_s16_mult_4_s16()
85 out_buff[2] = bias[out_ch + 2 + mult_tile]; in depthwise_conv_s16_mult_4_s16()
86 out_buff[3] = bias[out_ch + 3 + mult_tile]; in depthwise_conv_s16_mult_4_s16()
[all …]
|
| D | arm_depthwise_conv_s8.c | 83 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
84 ++in_ch, out_ch += ch_mult) in depthwise_conv_s8_mult_4()
99 … int32_t ker_idx = ker_h * (output_ch * kernel_x) + ker_w_start * output_ch + out_ch; in depthwise_conv_s8_mult_4()
|
| /cmsis-nn-latest/Tests/UnitTest/RefactoredTestGen/Lib/ |
| D | op_fully_connected.py | 45 out_ch = params["out_ch"]
48 shapes["weight_shape"] = (in_ch, 1, 1, out_ch)
51 shapes["bias_shape"] = [out_ch]
|
| D | op_conv.py | 52 out_ch = params["out_ch"]
58 if out_ch % groups != 0:
59 …timeError("ERROR: Output channels {} must be an even multiple of groups {}".format(out_ch, groups))
62 shapes["weight_shape"] = [params["filter_y"], params["filter_x"], filter_ch, out_ch]
65 shapes["bias_shape"] = [out_ch]
|
| /cmsis-nn-latest/Include/ |
| D | arm_nnsupportfunctions.h | 346 const int32_t out_ch,
384 const int32_t out_ch,
411 const int32_t out_ch,
|
