Home
last modified time | relevance | path

Searched refs:pooling_width (Results 1 – 25 of 38) sorted by relevance

12

/aosp_15_r20/external/XNNPACK/test/
H A Dmax-pooling-nhwc.cc25 .pooling_width(pool_size) in TEST()
45 .pooling_width(pool_size) in TEST()
65 .pooling_width(pool_size) in TEST()
82 .pooling_width(pool_size) in TEST()
101 .pooling_width(pool_size) in TEST()
118 .pooling_width(1) in TEST()
138 .pooling_width(1) in TEST()
158 .pooling_width(1) in TEST()
175 .pooling_width(1) in TEST()
194 .pooling_width(1) in TEST()
[all …]
H A Daverage-pooling-nhwc.cc42 .pooling_width(pooling_size.second) in TEST()
49 .pooling_width(pooling_size.first) in TEST()
69 .pooling_width(pooling_size.second) in TEST()
78 .pooling_width(pooling_size.first) in TEST()
101 .pooling_width(pooling_size.second) in TEST()
111 .pooling_width(pooling_size.first) in TEST()
134 .pooling_width(pooling_size.second) in TEST()
144 .pooling_width(pooling_size.first) in TEST()
165 .pooling_width(pooling_size.second) in TEST()
176 .pooling_width(pooling_size.second) in TEST()
[all …]
H A Dargmax-pooling-nhwc.cc38 .pooling_width(pool_size) in TEST()
58 .pooling_width(pool_size) in TEST()
78 .pooling_width(pool_size) in TEST()
95 .pooling_width(1) in TEST()
115 .pooling_width(1) in TEST()
135 .pooling_width(1) in TEST()
152 .pooling_width(1) in TEST()
161 .pooling_width(pool_size) in TEST()
178 .pooling_width(1) in TEST()
187 .pooling_width(pool_size) in TEST()
[all …]
H A Dunpooling-nhwc.cc20 .pooling_width(pooling_size) in TEST()
40 .pooling_width(pooling_size) in TEST()
58 .pooling_width(1) in TEST()
78 .pooling_width(1) in TEST()
96 .pooling_width(pooling_size) in TEST()
119 .pooling_width(3) in TEST()
139 .pooling_width(pooling_size) in TEST()
154 .pooling_width(3) in TEST()
169 .pooling_width(3) in TEST()
188 .pooling_width(3) in TEST()
[all …]
H A Dargmax-pooling-operator-tester.h93 const uint32_t total_padding_width = output_width() * pooling_width() - input_width(); in padding_left()
123 const uint32_t total_padding_width = output_width() * pooling_width() - input_width(); in padding_right()
185 …inline ArgmaxPoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width) { in pooling_size() argument
187 assert(pooling_width >= 1); in pooling_size()
189 this->pooling_width_ = pooling_width; in pooling_size()
203 inline ArgmaxPoolingOperatorTester& pooling_width(uint32_t pooling_width) { in pooling_width() function
204 assert(pooling_width >= 1); in pooling_width()
205 this->pooling_width_ = pooling_width; in pooling_width()
209 inline uint32_t pooling_width() const { in pooling_width() function
224 return (input_width() + pooling_width() - 1) / pooling_width(); in output_width()
[all …]
H A Dmax-pooling-2d.cc45 pooling_width = kernel_size_dist(rng); in MaxPooling2DTestBase()
48 padding_left = std::uniform_int_distribution<uint32_t>(0, pooling_width - 1)(rng); in MaxPooling2DTestBase()
49 padding_right = std::uniform_int_distribution<uint32_t>(0, pooling_width - 1)(rng); in MaxPooling2DTestBase()
54 stride_width = std::uniform_int_distribution<uint32_t>(1, pooling_width)(rng); in MaxPooling2DTestBase()
60 padding_left + input_width + padding_right, pooling_width, dilation_width, stride_width); in MaxPooling2DTestBase()
91 uint32_t pooling_width; member in MaxPooling2DTestBase
139 …ing_top, padding_right, padding_bottom, padding_left, pooling_height, pooling_width, stride_height, in TEST_F()
151 ASSERT_EQ(node->params.pooling_2d.pooling_width, pooling_width); in TEST_F()
190 …ing_top, padding_right, padding_bottom, padding_left, pooling_height, pooling_width, stride_height, in TEST_F()
203 ASSERT_EQ(node->params.pooling_2d.pooling_width, pooling_width); in TEST_F()
[all …]
H A Dunpooling-operator-tester.h143 inline UnpoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width) { in pooling_size() argument
145 assert(pooling_width >= 1); in pooling_size()
147 this->pooling_width_ = pooling_width; in pooling_size()
161 inline UnpoolingOperatorTester& pooling_width(uint32_t pooling_width) { in pooling_width() function
162 assert(pooling_width >= 1); in pooling_width()
163 this->pooling_width_ = pooling_width; in pooling_width()
167 inline uint32_t pooling_width() const { in pooling_width() function
178 return std::max<size_t>(input_width() * pooling_width(), padding_width) - padding_width; in output_width()
254 return std::max<size_t>(next_input_width() * pooling_width(), padding_width) - padding_width; in next_output_width()
284 …d::bind(std::uniform_int_distribution<uint32_t>(0, pooling_height() * pooling_width() - 1), std::r… in TestX32()
[all …]
H A Daverage-pooling-2d.cc35 pooling_width = pooling_size_dist(rng); in AveragePoolingTestF32()
38 input_padding_right = std::uniform_int_distribution<uint32_t>(0, pooling_width - 1)(rng); in AveragePoolingTestF32()
40 input_padding_left = std::uniform_int_distribution<uint32_t>(0, pooling_width - 1)(rng); in AveragePoolingTestF32()
46 input_padding_left + input_width + input_padding_right, pooling_width, 1, stride_width); in AveragePoolingTestF32()
66 uint32_t pooling_width; member in AveragePoolingTestF32
114 pooling_width, stride_height, stride_width, output_min, output_max, input_id, output_id, in TEST_F()
127 ASSERT_EQ(node->params.pooling_2d.pooling_width, pooling_width); in TEST_F()
151 …_top, input_padding_right, input_padding_bottom, input_padding_left, pooling_height, pooling_width, in TEST_F()
192 pooling_width, stride_height, stride_width, output_min, output_max, input_id, output_id, in TEST_F()
H A Daverage-pooling-operator-tester.h101 (output_width() - 1) * stride_width() + pooling_width() - input_width(); in padding_left()
133 (output_width() - 1) * stride_width() + pooling_width() - input_width(); in padding_right()
195 …inline AveragePoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width)… in pooling_size() argument
197 assert(pooling_width >= 1); in pooling_size()
199 this->pooling_width_ = pooling_width; in pooling_size()
213 inline AveragePoolingOperatorTester& pooling_width(uint32_t pooling_width) { in pooling_width() function
214 assert(pooling_width >= 1); in pooling_width()
215 this->pooling_width_ = pooling_width; in pooling_width()
219 inline uint32_t pooling_width() const { in pooling_width() function
276 if (padded_input_width <= pooling_width()) { in output_width()
[all …]
H A Dunpooling-2d.cc41 pooling_width = 2; in Unpooling2DTestBase()
44 …= xnn_compute_unpooling_output_dimension(input_width, padding_left + padding_right, pooling_width); in Unpooling2DTestBase()
46 index_dist = std::uniform_int_distribution<uint32_t>(0, pooling_height * pooling_width - 1); in Unpooling2DTestBase()
79 uint32_t pooling_width; member in Unpooling2DTestBase
127 pooling_width, input_value_id, input_index_id, output_id, in TEST_F()
139 ASSERT_EQ(node->params.pooling_2d.pooling_width, pooling_width); in TEST_F()
161 …padding_top, padding_right, padding_bottom, padding_left, pooling_height, pooling_width, channels,… in TEST_F()
207 pooling_width, input_value_id, input_index_id, output_id, in TEST_F()
H A Dmax-pooling-operator-tester.h194 inline MaxPoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width) { in pooling_size() argument
196 assert(pooling_width >= 1); in pooling_size()
198 this->pooling_width_ = pooling_width; in pooling_size()
212 inline MaxPoolingOperatorTester& pooling_width(uint32_t pooling_width) { in pooling_width() argument
213 assert(pooling_width >= 1); in pooling_width()
214 this->pooling_width_ = pooling_width; in pooling_width()
218 inline uint32_t pooling_width() const { in pooling_width() function
297 return (pooling_width() - 1) * dilation_width() + 1; in dilated_pooling_width()
472 for (size_t px = 0; px < pooling_width(); px++) { in TestS8()
496 pooling_height(), pooling_width(), in TestS8()
[all …]
H A Dargmax-pooling-2d.cc41 pooling_width = pooling_size_dist(rng); in ArgmaxPoolingTestF32()
47 … = compute_output_dimension(input_width + input_padding_left + input_padding_right, pooling_width); in ArgmaxPoolingTestF32()
66 uint32_t pooling_width; member in ArgmaxPoolingTestF32
120 pooling_height, pooling_width, input_id, output_value_id, output_index_id, in TEST_F()
132 ASSERT_EQ(node->params.pooling_2d.pooling_width, pooling_width); in TEST_F()
153 …_top, input_padding_right, input_padding_bottom, input_padding_left, pooling_height, pooling_width, in TEST_F()
201 pooling_height, pooling_width, input_id, output_value_id, output_index_id, in TEST_F()
/aosp_15_r20/external/XNNPACK/src/operators/
H A Dmax-pooling-nhwc.c43 uint32_t pooling_width, in create_max_pooling2d_nhwc() argument
78 const uint32_t pooling_size = pooling_height * pooling_width; in create_max_pooling2d_nhwc()
84 pooling_width, pooling_height); in create_max_pooling2d_nhwc()
116 if (stride_width > pooling_width) { in create_max_pooling2d_nhwc()
119 xnn_operator_type_to_string(operator_type), stride_width, pooling_width); in create_max_pooling2d_nhwc()
174 max_pooling_op->kernel_width = pooling_width; in create_max_pooling2d_nhwc()
273 const size_t pooling_width = max_pooling_op->kernel_width; in setup_max_pooling2d_nhwc() local
274 const size_t pooling_size = pooling_height * pooling_width; in setup_max_pooling2d_nhwc()
280 …max_pooling_op->dilation_width > 1 ? pooling_width : min(max_pooling_op->stride_width, pooling_wid… in setup_max_pooling2d_nhwc()
343 uint32_t pooling_width, in xnn_create_max_pooling2d_nhwc_s8() argument
[all …]
H A Daverage-pooling-nhwc.c43 uint32_t pooling_width, in xnn_create_average_pooling2d_nhwc_qu8() argument
69 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_create_average_pooling2d_nhwc_qu8()
74 …xnn_operator_type_to_string(xnn_operator_type_average_pooling_nhwc_qu8), pooling_width, pooling_he… in xnn_create_average_pooling2d_nhwc_qu8()
99 if (stride_width > pooling_width) { in xnn_create_average_pooling2d_nhwc_qu8()
102 …_operator_type_to_string(xnn_operator_type_average_pooling_nhwc_qu8), stride_width, pooling_width); in xnn_create_average_pooling2d_nhwc_qu8()
179 pooling_size, pooling_width, pooling_height); in xnn_create_average_pooling2d_nhwc_qu8()
210 average_pooling_op->kernel_width = pooling_width; in xnn_create_average_pooling2d_nhwc_qu8()
251 uint32_t pooling_width, in xnn_create_average_pooling2d_nhwc_f16() argument
282 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_create_average_pooling2d_nhwc_f16()
287 …xnn_operator_type_to_string(xnn_operator_type_average_pooling_nhwc_f16), pooling_width, pooling_he… in xnn_create_average_pooling2d_nhwc_f16()
[all …]
H A Dargmax-pooling-nhwc.c48 uint32_t pooling_width, in xnn_create_argmax_pooling2d_nhwc_f32() argument
66 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_create_argmax_pooling2d_nhwc_f32()
71 …xnn_operator_type_to_string(xnn_operator_type_argmax_pooling_nhwc_f32), pooling_width, pooling_hei… in xnn_create_argmax_pooling2d_nhwc_f32()
133 argmax_pooling_op->kernel_width = pooling_width; in xnn_create_argmax_pooling2d_nhwc_f32()
135 argmax_pooling_op->stride_width = pooling_width; in xnn_create_argmax_pooling2d_nhwc_f32()
197 const size_t pooling_width = argmax_pooling_op->kernel_width; in xnn_setup_argmax_pooling2d_nhwc_f32() local
201 argmax_pooling_op->output_width = divide_round_up(input_width, pooling_width); in xnn_setup_argmax_pooling2d_nhwc_f32()
204 const uint32_t padding_width = argmax_pooling_op->output_width * pooling_width - input_width; in xnn_setup_argmax_pooling2d_nhwc_f32()
218 const size_t pooling_size = pooling_height * pooling_width; in xnn_setup_argmax_pooling2d_nhwc_f32()
224 const size_t step_width = pooling_width; in xnn_setup_argmax_pooling2d_nhwc_f32()
H A Dunpooling-nhwc.c30 uint32_t pooling_width, in xnn_create_unpooling2d_nhwc_x32() argument
48 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_create_unpooling2d_nhwc_x32()
53 … xnn_operator_type_to_string(xnn_operator_type_unpooling_nhwc_x32), pooling_width, pooling_height); in xnn_create_unpooling2d_nhwc_x32()
103 unpooling_op->kernel_width = pooling_width; in xnn_create_unpooling2d_nhwc_x32()
184 const size_t pooling_width = unpooling_op->kernel_width; in xnn_setup_unpooling2d_nhwc_x32() local
185 const size_t pooling_size = pooling_height * pooling_width; in xnn_setup_unpooling2d_nhwc_x32()
/aosp_15_r20/external/executorch/backends/xnnpack/operators/
H A Dop_max_dim.py65 pooling_width = 1
70 pooling_width = 1
74 pooling_width = input_shape[2]
84 pooling_width=pooling_width,
182 pooling_width = 1
185 pooling_width = 1
188 pooling_width = input_shape[2]
197 pooling_width=pooling_width,
H A Dop_avg_pooling2d.py47 pooling_height, pooling_width = cast(List, node.args[1])
64 pooling_width=pooling_width,
/aosp_15_r20/external/XNNPACK/src/subgraph/
H A Daverage-pooling-2d.c48 node->params.pooling_2d.pooling_width, in create_average_pooling_operator()
65 node->params.pooling_2d.pooling_width, in create_average_pooling_operator()
142 uint32_t pooling_width, in xnn_define_average_pooling_2d() argument
156 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_define_average_pooling_2d()
161 xnn_node_type_to_string(xnn_node_type_average_pooling_2d), pooling_width, pooling_height); in xnn_define_average_pooling_2d()
187 if (stride_width > pooling_width) { in xnn_define_average_pooling_2d()
190 xnn_node_type_to_string(xnn_node_type_max_pooling_2d), stride_width, pooling_width); in xnn_define_average_pooling_2d()
267 node->params.pooling_2d.pooling_width = pooling_width; in xnn_define_average_pooling_2d()
H A Dmax-pooling-2d.c50 node->params.pooling_2d.pooling_width, in create_max_pooling_operator()
69 node->params.pooling_2d.pooling_width, in create_max_pooling_operator()
93 node->params.pooling_2d.pooling_width, in create_max_pooling_operator()
119 node->params.pooling_2d.pooling_width, in create_max_pooling_operator()
222 uint32_t pooling_width, in xnn_define_max_pooling_2d() argument
238 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_define_max_pooling_2d()
243 xnn_node_type_to_string(xnn_node_type_max_pooling_2d), pooling_width, pooling_height); in xnn_define_max_pooling_2d()
275 if (stride_width > pooling_width) { in xnn_define_max_pooling_2d()
278 xnn_node_type_to_string(xnn_node_type_max_pooling_2d), stride_width, pooling_width); in xnn_define_max_pooling_2d()
400 node->params.pooling_2d.pooling_width = pooling_width; in xnn_define_max_pooling_2d()
H A Dunpooling-2d.c50 node->params.pooling_2d.pooling_width, in create_unpooling_operator()
113 uint32_t pooling_width, in xnn_define_unpooling_2d() argument
124 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_define_unpooling_2d()
129 xnn_node_type_to_string(xnn_node_type_unpooling_2d), pooling_width, pooling_height); in xnn_define_unpooling_2d()
213 node->params.pooling_2d.pooling_width = pooling_width; in xnn_define_unpooling_2d()
H A Dargmax-pooling-2d.c51 node->params.pooling_2d.pooling_width, in create_argmax_pooling_operator()
114 uint32_t pooling_width, in xnn_define_argmax_pooling_2d() argument
125 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_define_argmax_pooling_2d()
130 xnn_node_type_to_string(xnn_node_type_argmax_pooling_2d), pooling_width, pooling_height); in xnn_define_argmax_pooling_2d()
216 node->params.pooling_2d.pooling_width = pooling_width; in xnn_define_argmax_pooling_2d()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
H A Daverage-pooling.c36 uint32_t pooling_width, in pytorch_qnnp_create_average_pooling2d_nhwc_q8() argument
59 const uint32_t pooling_size = pooling_height * pooling_width; in pytorch_qnnp_create_average_pooling2d_nhwc_q8()
65 pooling_width, in pytorch_qnnp_create_average_pooling2d_nhwc_q8()
131 pooling_width, in pytorch_qnnp_create_average_pooling2d_nhwc_q8()
167 average_pooling->kernel_width = pooling_width; in pytorch_qnnp_create_average_pooling2d_nhwc_q8()
176 size_t nrows = pooling_height * pooling_width; in pytorch_qnnp_create_average_pooling2d_nhwc_q8()
H A Dmax-pooling.c41 uint32_t pooling_width, in pytorch_qnnp_create_max_pooling2d_nhwc_u8() argument
62 const uint32_t pooling_size = pooling_height * pooling_width; in pytorch_qnnp_create_max_pooling2d_nhwc_u8()
68 pooling_width, in pytorch_qnnp_create_max_pooling2d_nhwc_u8()
122 max_pooling->kernel_width = pooling_width; in pytorch_qnnp_create_max_pooling2d_nhwc_u8()
/aosp_15_r20/external/XNNPACK/src/
H A Dindirection.c273 const size_t pooling_width = op->kernel_width; in xnn_indirection_init_maxpool2d() local
301 for (size_t pooling_x = 0; pooling_x < pooling_width; pooling_x++) { in xnn_indirection_init_maxpool2d()
326 for (size_t pooling_x = 0; pooling_x < pooling_width; pooling_x++) { in xnn_indirection_init_maxpool2d()
838 const size_t pooling_width = op->kernel_width; in xnn_indirection_init_unpool2d() local
847 for (size_t pooling_x = 0; pooling_x < pooling_width; pooling_x++) { in xnn_indirection_init_unpool2d()
848 …const size_t output_x = min(doz(input_x * pooling_width + pooling_x, output_padding_left), output_… in xnn_indirection_init_unpool2d()
849 …_buffer[(((image * input_height + input_y) * input_width + input_x) * pooling_width + pooling_x) *… in xnn_indirection_init_unpool2d()
864 size_t pooling_width, in xnn_indirection_init_pavgpool2d_f16() argument
877 …const size_t input_x_end = min(doz(output_x * stride_width + pooling_width, padding_left), input_w… in xnn_indirection_init_pavgpool2d_f16()
890 size_t pooling_width, in xnn_indirection_init_pavgpool2d_f32() argument
[all …]

12