Searched refs:dilation_y (Results 1 – 12 of 12) sorted by relevance
| /cmsis-nn-latest/Source/ConvolutionFunctions/ |
| D | arm_depthwise_conv_s4.c | 68 const int32_t dilation_y) in depthwise_conv_s4_generic()
115 if (dilation_y > 1) in depthwise_conv_s4_generic()
117 … const int32_t start_y_max = (-base_idx_y + dilation_y - 1) / dilation_y; in depthwise_conv_s4_generic()
119 … const int32_t end_min_y = (input_y - base_idx_y + dilation_y - 1) / dilation_y; in depthwise_conv_s4_generic()
134 int32_t idx_y = base_idx_y + dilation_y * ker_y_start; in depthwise_conv_s4_generic()
157 idx_y += dilation_y; in depthwise_conv_s4_generic()
207 if (dilation_y > 1) in depthwise_conv_s4_generic()
209 … const int32_t start_y_max = (-base_idx_y + dilation_y - 1) / dilation_y; in depthwise_conv_s4_generic()
211 … const int32_t end_min_y = (input_y - base_idx_y + dilation_y - 1) / dilation_y; in depthwise_conv_s4_generic()
225 int32_t idx_y = base_idx_y + dilation_y * ker_y_start; in depthwise_conv_s4_generic()
[all …]
|
| D | arm_depthwise_conv_s16.c | 155 const uint16_t dilation_y) in depthwise_conv_s16_generic_s16() argument
193 if (dilation_y > 1) in depthwise_conv_s16_generic_s16()
195 const int32_t start_y_max = (-base_idx_y + dilation_y - 1) / dilation_y; in depthwise_conv_s16_generic_s16()
197 … const int32_t end_min_y = (input_y - base_idx_y + dilation_y - 1) / dilation_y; in depthwise_conv_s16_generic_s16()
213 const int32_t idx_y = base_idx_y + dilation_y * i_ker_y; in depthwise_conv_s16_generic_s16()
257 const uint16_t dilation_y = dw_conv_params->dilation.h; in arm_depthwise_conv_s16() local
284 dilation_y); in arm_depthwise_conv_s16()
|
| D | arm_depthwise_conv_s8.c | 179 const uint16_t dilation_y) in depthwise_conv_s8_generic()
219 if (dilation_y > 1) in depthwise_conv_s8_generic()
221 const int32_t start_y_max = (-base_idx_y + dilation_y - 1) / dilation_y; in depthwise_conv_s8_generic()
223 … const int32_t end_min_y = (input_y - base_idx_y + dilation_y - 1) / dilation_y; in depthwise_conv_s8_generic()
239 const int32_t idx_y = base_idx_y + dilation_y * i_ker_y; in depthwise_conv_s8_generic()
286 const uint16_t dilation_y = dw_conv_params->dilation.h; in arm_depthwise_conv_s8() local
345 dilation_y); in arm_depthwise_conv_s8()
|
| D | arm_convolve_s4.c | 85 const int32_t dilation_y = conv_params->dilation.h; in arm_convolve_s4() local
117 const int32_t k_y = base_idx_y + dilation_y * i_ker_y; in arm_convolve_s4()
201 const int32_t k_y = base_idx_y + dilation_y * i_ker_y; in arm_convolve_s4()
|
| D | arm_convolve_s16.c | 82 const int32_t dilation_y = conv_params->dilation.h; in arm_convolve_s16() local
116 const int32_t k_y = base_idx_y + dilation_y * i_ker_y; in arm_convolve_s16()
|
| D | arm_convolve_s8.c | 85 const int32_t dilation_y = conv_params->dilation.h; in arm_convolve_s8() local
141 const int32_t k_y = base_idx_y + dilation_y * i_ker_y; in arm_convolve_s8()
|
| /cmsis-nn-latest/Tests/UnitTest/TestCases/Common/ |
| D | conv2d_s4_weights_template.json | 114 "dilation_h_factor": dilation_y
|
| D | dw_s4_weights_template.json | 108 "dilation_h_factor": dilation_y
|
| /cmsis-nn-latest/Tests/UnitTest/ |
| D | conv_settings.py | 56 dilation_y=1,
87 dilation_y=dilation_y,
139 f.write("#define {}_DILATION_Y {}\n".format(prefix, self.dilation_y))
268 dilation_filter_y = (self.filter_y - 1) * (self.dilation_y - 1)
294 "dilation_y": self.dilation_y,
343 dilation_rate=(self.dilation_y, self.dilation_x),
357 … dilation_rate=(self.dilation_y, self.dilation_x),
370 dilation_rate=(self.dilation_y,
|
| D | generate_test_data.py | 484 dilation_y=2,
505 dilation_y=2,
526 dilation_y=2,
547 dilation_y=2,
568 dilation_y=2,
589 dilation_y=2,
710 dilation_y=2,
734 dilation_y=2,
856 dilation_y=2,
881 dilation_y=2,
[all …]
|
| D | test_settings.py | 85 dilation_y=1,
155 self.dilation_y = dilation_y
405 filter_y = (self.filter_y - 1) * self.dilation_y + 1
|
| D | model_extractor.py | 130 f.write("#define {}_DILATION_Y {}\n".format(prefix, self.dilation_y))
132 f.write("#define {}_DILATION_H {}\n".format(prefix, self.dilation_y))
|
