| /aosp_15_r20/external/XNNPACK/test/ |
| H A D | convolution-test-helpers.cc | 28 size_t subsampling_width, in compute_convolution_qs8_reference_results() argument
63 const size_t ix = ox * subsampling_width + kx * dilation_width - input_padding_left; in compute_convolution_qs8_reference_results()
98 size_t subsampling_width, in compute_convolution_qs8_reference_results() argument
124 subsampling_width, in compute_convolution_qs8_reference_results()
152 size_t subsampling_width, in compute_convolution_qu8_reference_results() argument
188 const size_t ix = ox * subsampling_width + kx * dilation_width - input_padding_left; in compute_convolution_qu8_reference_results()
222 size_t subsampling_width, in compute_convolution_qu8_reference_results() argument
249 subsampling_width, in compute_convolution_qu8_reference_results()
278 size_t subsampling_width, in compute_depthwise_convolution_qs8_reference_results() argument
312 const size_t ix = ox * subsampling_width + kx * dilation_width - input_padding_left; in compute_depthwise_convolution_qs8_reference_results()
[all …]
|
| H A D | convolution-2d.cc | 44 subsampling_width = subsampling_height; in ConvolutionTestBase()
53 …compute_convolution_output_dimension(input_width, kernel_width, dilation_width, subsampling_width); in ConvolutionTestBase()
87 uint32_t subsampling_width; member in xnnpack::ConvolutionTestBase
174 kernel_width, subsampling_height, subsampling_width, dilation_height, dilation_width, groups, in TEST_F()
189 ASSERT_EQ(node->params.convolution_2d.subsampling_width, subsampling_width); in TEST_F()
244 kernel_width, subsampling_height, subsampling_width, dilation_height, dilation_width, groups, in TEST_F()
259 ASSERT_EQ(node->params.convolution_2d.subsampling_width, subsampling_width); in TEST_F()
314 kernel_width, subsampling_height, subsampling_width, dilation_height, dilation_width, groups, in TEST_F()
329 ASSERT_EQ(node->params.convolution_2d.subsampling_width, subsampling_width); in TEST_F()
385 kernel_width, subsampling_height, subsampling_width, dilation_height, dilation_width, groups, in TEST_F()
[all …]
|
| H A D | depthwise-convolution-2d.cc | 45 subsampling_width = stride_dist(rng); in DepthwiseConvolutionTestBase()
56 …padding_left + input_width + input_padding_right, kernel_width, dilation_width, subsampling_width); in DepthwiseConvolutionTestBase()
91 uint32_t subsampling_width; member in xnnpack::DepthwiseConvolutionTestBase
201 …kernel_width, subsampling_height, subsampling_width, dilation_height, dilation_width, depth_multip… in TEST_F()
216 ASSERT_EQ(node->params.depthwise_convolution_2d.subsampling_width, subsampling_width); in TEST_F()
274 …kernel_width, subsampling_height, subsampling_width, dilation_height, dilation_width, depth_multip… in TEST_F()
289 ASSERT_EQ(node->params.depthwise_convolution_2d.subsampling_width, subsampling_width); in TEST_F()
347 …kernel_width, subsampling_height, subsampling_width, dilation_height, dilation_width, depth_multip… in TEST_F()
362 ASSERT_EQ(node->params.depthwise_convolution_2d.subsampling_width, subsampling_width); in TEST_F()
417 …kernel_width, subsampling_height, subsampling_width, dilation_height, dilation_width, depth_multip… in TEST_F()
[all …]
|
| H A D | convolution-test-helpers.h | 26 size_t subsampling_width,
53 size_t subsampling_width,
79 size_t subsampling_width,
106 size_t subsampling_width,
134 size_t subsampling_width,
160 size_t subsampling_width,
185 size_t subsampling_width,
212 size_t subsampling_width,
|
| H A D | convolution-operator-tester.h | 109 (output_width() - 1) * subsampling_width() + dilated_kernel_width() - input_width(); in padding_left()
141 (output_width() - 1) * subsampling_width() + dilated_kernel_width() - input_width(); in padding_right()
293 …e ConvolutionOperatorTester& subsampling(uint32_t subsampling_height, uint32_t subsampling_width) { in subsampling() argument
295 assert(subsampling_width >= 1); in subsampling()
297 this->subsampling_width_ = subsampling_width; in subsampling()
311 inline ConvolutionOperatorTester& subsampling_width(uint32_t subsampling_width) { in subsampling_width() argument
312 assert(subsampling_width >= 1); in subsampling_width()
313 this->subsampling_width_ = subsampling_width; in subsampling_width()
317 inline uint32_t subsampling_width() const { in subsampling_width() function
374 return (input_width() + subsampling_width() - 1) / subsampling_width(); in output_width()
[all …]
|
| H A D | conv-hwc-microkernel-tester.h | 204 inline ConvHWCMicrokernelTester& subsampling_width(uint32_t subsampling_width) { in subsampling_width() function
205 assert(subsampling_width >= 1); in subsampling_width()
206 this->subsampling_width_ = subsampling_width; in subsampling_width()
210 inline uint32_t subsampling_width() const { in subsampling_width() function
252 …return (std::max<size_t>(padded_input_width + subsampling_width(), kernel_width()) - kernel_width(… in output_width()
253 / subsampling_width(); in output_width()
327 const size_t ix = ox * subsampling_width() + kx - padding_left();
|
| H A D | conv-hwc2chw-microkernel-tester.h | 206 inline ConvHWC2CHWMicrokernelTester& subsampling_width(uint32_t subsampling_width) { in subsampling_width() argument
207 assert(subsampling_width >= 1); in subsampling_width()
208 this->subsampling_width_ = subsampling_width; in subsampling_width()
212 inline uint32_t subsampling_width() const { in subsampling_width() function
260 return (padded_input_width - kernel_width()) / subsampling_width() + 1; in output_width()
335 const size_t ix = ox * subsampling_width() + kx - padding_left();
450 const size_t ix = ox * subsampling_width() + kx - padding_left(); in Test()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/ |
| H A D | convolution.c | 52 uint32_t subsampling_width, in pytorch_qnnp_create_convolution_ndhwc_q8() argument
91 if (subsampling_width == 0 || subsampling_height == 0) { in pytorch_qnnp_create_convolution_ndhwc_q8()
96 subsampling_width, in pytorch_qnnp_create_convolution_ndhwc_q8()
121 subsampling_width, in pytorch_qnnp_create_convolution_ndhwc_q8()
125 if (subsampling_width > kernel_width) { in pytorch_qnnp_create_convolution_ndhwc_q8()
133 subsampling_width, in pytorch_qnnp_create_convolution_ndhwc_q8()
212 kernel_size == 1 && subsampling_height == 1 && subsampling_width == 1 && in pytorch_qnnp_create_convolution_ndhwc_q8()
511 convolution->stride_width = subsampling_width; in pytorch_qnnp_create_convolution_ndhwc_q8()
555 uint32_t subsampling_width, in pytorch_qnnp_create_convolution2d_nhwc_q8() argument
581 subsampling_width, in pytorch_qnnp_create_convolution2d_nhwc_q8()
[all …]
|
| /aosp_15_r20/external/XNNPACK/src/subgraph/ |
| H A D | depthwise-convolution-2d.c | 66 node->params.depthwise_convolution_2d.subsampling_width, in create_convolution_operator()
94 node->params.depthwise_convolution_2d.subsampling_width, in create_convolution_operator()
120 node->params.depthwise_convolution_2d.subsampling_width, in create_convolution_operator()
152 node->params.depthwise_convolution_2d.subsampling_width, in create_convolution_operator()
186 node->params.depthwise_convolution_2d.subsampling_width, in create_convolution_operator()
222 node->params.depthwise_convolution_2d.subsampling_width, in create_convolution_operator()
446 uint32_t subsampling_width, in xnn_define_depthwise_convolution_2d() argument
471 if (subsampling_width == 0 || subsampling_height == 0) { in xnn_define_depthwise_convolution_2d()
474 …xnn_node_type_to_string(xnn_node_type_depthwise_convolution_2d), subsampling_width, subsampling_he… in xnn_define_depthwise_convolution_2d()
524 …if ((flags & XNN_FLAG_TENSORFLOW_SAME_PADDING) != 0 && (subsampling_height | subsampling_width) ==… in xnn_define_depthwise_convolution_2d()
[all …]
|
| H A D | convolution-2d.c | 65 node->params.convolution_2d.subsampling_width, in create_convolution_operator()
93 node->params.convolution_2d.subsampling_width, in create_convolution_operator()
119 node->params.convolution_2d.subsampling_width, in create_convolution_operator()
151 node->params.convolution_2d.subsampling_width, in create_convolution_operator()
185 node->params.convolution_2d.subsampling_width, in create_convolution_operator()
221 node->params.convolution_2d.subsampling_width, in create_convolution_operator()
445 uint32_t subsampling_width, in xnn_define_convolution_2d() argument
471 if (subsampling_width == 0 || subsampling_height == 0) { in xnn_define_convolution_2d()
474 xnn_node_type_to_string(xnn_node_type_convolution_2d), subsampling_width, subsampling_height); in xnn_define_convolution_2d()
531 …if ((flags & XNN_FLAG_TENSORFLOW_SAME_PADDING) != 0 && (subsampling_height | subsampling_width) ==… in xnn_define_convolution_2d()
[all …]
|
| /aosp_15_r20/external/XNNPACK/src/operators/ |
| H A D | convolution-nchw.c | 35 uint32_t subsampling_width, in xnn_create_convolution2d_nchw_f32() argument
69 if (subsampling_width == 0 || subsampling_height == 0) { in xnn_create_convolution2d_nchw_f32()
72 …xnn_operator_type_to_string(xnn_operator_type_convolution_nchw_f32), subsampling_width, subsamplin… in xnn_create_convolution2d_nchw_f32()
165 …1x1 = kernel_width == 1 && kernel_height == 1 && subsampling_height == 1 && subsampling_width == 1; in xnn_create_convolution2d_nchw_f32()
171 } else if (is_3x3 && subsampling_height == 2 && subsampling_width == 2 && in xnn_create_convolution2d_nchw_f32()
176 } else if (is_3x3 && subsampling_height == 1 && subsampling_width == 1 && in xnn_create_convolution2d_nchw_f32()
182 } else if (is_3x3 && subsampling_height == 2 && subsampling_width == 2 && in xnn_create_convolution2d_nchw_f32()
188 } else if (is_5x5 && subsampling_height == 1 && subsampling_width == 1 && in xnn_create_convolution2d_nchw_f32()
194 } else if (is_5x5 && subsampling_height == 2 && subsampling_width == 2 && in xnn_create_convolution2d_nchw_f32()
206 …kernel_width, kernel_height, subsampling_width, subsampling_height, dilation_width, dilation_heigh… in xnn_create_convolution2d_nchw_f32()
[all …]
|
| H A D | convolution-nhwc.c | 205 uint32_t subsampling_width, in create_convolution2d_nhwc() argument
278 if (subsampling_width == 0 || subsampling_height == 0) { in create_convolution2d_nhwc()
281 xnn_operator_type_to_string(operator_type), subsampling_width, subsampling_height); in create_convolution2d_nhwc()
371 const bool unit_subsampling = (subsampling_width | subsampling_height) == 1; in create_convolution2d_nhwc()
627 convolution_op->stride_width = subsampling_width; in create_convolution2d_nhwc()
661 uint32_t subsampling_width, in xnn_create_convolution2d_nhwc_qu8() argument
744 subsampling_height, subsampling_width, in xnn_create_convolution2d_nhwc_qu8()
780 uint32_t subsampling_width, in xnn_create_convolution2d_nhwc_qs8() argument
858 subsampling_height, subsampling_width, in xnn_create_convolution2d_nhwc_qs8()
894 uint32_t subsampling_width, in xnn_create_convolution2d_nhwc_qc8() argument
[all …]
|
| /aosp_15_r20/external/XNNPACK/src/ |
| H A D | im2col.c | 19 size_t subsampling_width, in xnn_im2col_conv2d() argument
36 …const size_t input_x = output_x * subsampling_width + kernel_x * dilation_width - input_padding_le… in xnn_im2col_conv2d()
|
| H A D | subgraph.c | 263 …->params.convolution_2d.subsampling_height | node->params.convolution_2d.subsampling_width) != 1) { in xnn_check_nchw_compatibility()
268 node->params.convolution_2d.subsampling_width); in xnn_check_nchw_compatibility()
284 …->params.convolution_2d.subsampling_height | node->params.convolution_2d.subsampling_width) != 2) { in xnn_check_nchw_compatibility()
289 node->params.convolution_2d.subsampling_width); in xnn_check_nchw_compatibility()
328 …ise_convolution_2d.subsampling_height != node->params.depthwise_convolution_2d.subsampling_width) { in xnn_check_nchw_compatibility()
333 node->params.depthwise_convolution_2d.subsampling_width); in xnn_check_nchw_compatibility()
|
| /aosp_15_r20/external/XNNPACK/src/xnnpack/ |
| H A D | subgraph.h | 188 uint32_t subsampling_width; member
220 uint32_t subsampling_width; member
|
| H A D | im2col.h | 22 size_t subsampling_width,
|
| /aosp_15_r20/external/XNNPACK/include/ |
| H A D | xnnpack.h | 356 uint32_t subsampling_width,
467 uint32_t subsampling_width,
1647 uint32_t subsampling_width,
1679 uint32_t subsampling_width,
2104 uint32_t subsampling_width,
2403 uint32_t subsampling_width,
2901 uint32_t subsampling_width,
2966 uint32_t subsampling_width,
3273 uint32_t subsampling_width,
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/include/ |
| H A D | pytorch_qnnpack.h | 54 uint32_t subsampling_width,
81 uint32_t subsampling_width,
|
| /aosp_15_r20/external/executorch/backends/xnnpack/serialization/ |
| H A D | xnnpack_graph_schema.py | 57 subsampling_width: int
|
| H A D | schema.fbs | 233 subsampling_width:uint;
|
| H A D | runtime_schema.fbs | 237 subsampling_width:uint;
|
| /aosp_15_r20/external/executorch/backends/xnnpack/runtime/ |
| H A D | XNNCompiler.cpp | 959 graph_node->subsampling_width(), in defineConv2dNode()
1180 graph_node->subsampling_width(), in defineDepthwiseConv2dNode()
|