Searched refs:dilation_x (Results 1 – 12 of 12) sorted by relevance
| /cmsis-nn-3.7.0/Source/ConvolutionFunctions/ |
| D | arm_depthwise_conv_s4.c | 67 const int32_t dilation_x, in depthwise_conv_s4_generic()
102 if (dilation_x > 1) in depthwise_conv_s4_generic()
104 … const int32_t start_x_max = (-base_idx_x + dilation_x - 1) / dilation_x; in depthwise_conv_s4_generic()
106 … const int32_t end_min_x = (input_x - base_idx_x + dilation_x - 1) / dilation_x; in depthwise_conv_s4_generic()
137 int32_t idx_x = base_idx_x + dilation_x * ker_x_start; in depthwise_conv_s4_generic()
153 idx_0 += dilation_x * input_ch; in depthwise_conv_s4_generic()
154 idx_x += dilation_x; in depthwise_conv_s4_generic()
194 if (dilation_x > 1) in depthwise_conv_s4_generic()
196 … const int32_t start_x_max = (-base_idx_x + dilation_x - 1) / dilation_x; in depthwise_conv_s4_generic()
198 … const int32_t end_min_x = (input_x - base_idx_x + dilation_x - 1) / dilation_x; in depthwise_conv_s4_generic()
[all …]
|
| D | arm_depthwise_conv_s16.c | 154 const uint16_t dilation_x, in depthwise_conv_s16_generic_s16() argument
180 if (dilation_x > 1) in depthwise_conv_s16_generic_s16()
182 const int32_t start_x_max = (-base_idx_x + dilation_x - 1) / dilation_x; in depthwise_conv_s16_generic_s16()
184 … const int32_t end_min_x = (input_x - base_idx_x + dilation_x - 1) / dilation_x; in depthwise_conv_s16_generic_s16()
216 const int32_t idx_x = base_idx_x + dilation_x * i_ker_x; in depthwise_conv_s16_generic_s16()
256 const uint16_t dilation_x = dw_conv_params->dilation.w; in arm_depthwise_conv_s16() local
283 dilation_x, in arm_depthwise_conv_s16()
|
| D | arm_depthwise_conv_s8.c | 178 const uint16_t dilation_x, in depthwise_conv_s8_generic()
206 if (dilation_x > 1) in depthwise_conv_s8_generic()
208 const int32_t start_x_max = (-base_idx_x + dilation_x - 1) / dilation_x; in depthwise_conv_s8_generic()
210 … const int32_t end_min_x = (input_x - base_idx_x + dilation_x - 1) / dilation_x; in depthwise_conv_s8_generic()
242 const int32_t idx_x = base_idx_x + dilation_x * i_ker_x; in depthwise_conv_s8_generic()
285 const uint16_t dilation_x = dw_conv_params->dilation.w; in arm_depthwise_conv_s8() local
344 dilation_x, in arm_depthwise_conv_s8()
|
| D | arm_convolve_s4.c | 84 const int32_t dilation_x = conv_params->dilation.w; in arm_convolve_s4() local
118 const int32_t k_x = base_idx_x + dilation_x * i_ker_x; in arm_convolve_s4()
202 const int32_t k_x = base_idx_x + dilation_x * i_ker_x; in arm_convolve_s4()
|
| D | arm_convolve_s16.c | 81 const int32_t dilation_x = conv_params->dilation.w; in arm_convolve_s16() local
117 const int32_t k_x = base_idx_x + dilation_x * i_ker_x; in arm_convolve_s16()
|
| D | arm_convolve_s8.c | 84 const int32_t dilation_x = conv_params->dilation.w; in arm_convolve_s8() local
142 const int32_t k_x = base_idx_x + dilation_x * i_ker_x; in arm_convolve_s8()
|
| /cmsis-nn-3.7.0/Tests/UnitTest/TestCases/Common/ |
| D | conv2d_s4_weights_template.json | 113 "dilation_w_factor": dilation_x,
|
| D | dw_s4_weights_template.json | 107 "dilation_w_factor": dilation_x,
|
| /cmsis-nn-3.7.0/Tests/UnitTest/ |
| D | conv_settings.py | 55 dilation_x=1,
86 dilation_x=dilation_x,
138 f.write("#define {}_DILATION_X {}\n".format(prefix, self.dilation_x))
267 dilation_filter_x = (self.filter_x - 1) * (self.dilation_x - 1)
293 "dilation_x": self.dilation_x,
343 dilation_rate=(self.dilation_y, self.dilation_x),
357 … dilation_rate=(self.dilation_y, self.dilation_x),
371 … self.dilation_x),
|
| D | generate_test_data.py | 483 dilation_x=3,
504 dilation_x=2,
525 dilation_x=2,
546 dilation_x=3,
567 dilation_x=2,
588 dilation_x=2,
709 dilation_x=2,
733 dilation_x=2,
757 dilation_x=2,
855 dilation_x=2,
[all …]
|
| D | test_settings.py | 84 dilation_x=1,
154 self.dilation_x = dilation_x
404 filter_x = (self.filter_x - 1) * self.dilation_x + 1
|
| D | model_extractor.py | 129 f.write("#define {}_DILATION_X {}\n".format(prefix, self.dilation_x))
131 f.write("#define {}_DILATION_W {}\n".format(prefix, self.dilation_x))
|
