Home
last modified time | relevance | path

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

/external/XNNPACK/test/
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 …]
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 …]
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 …]
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 …]
Dargmax-pooling-operator-tester.h94 const uint32_t total_padding_width = output_width() * pooling_width() - input_width(); in padding_left()
124 const uint32_t total_padding_width = output_width() * pooling_width() - input_width(); in padding_right()
186 …inline ArgmaxPoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width) { in pooling_size() argument
188 assert(pooling_width >= 1); in pooling_size()
190 this->pooling_width_ = pooling_width; in pooling_size()
204 inline ArgmaxPoolingOperatorTester& pooling_width(uint32_t pooling_width) { in pooling_width() argument
205 assert(pooling_width >= 1); in pooling_width()
206 this->pooling_width_ = pooling_width; in pooling_width()
210 inline uint32_t pooling_width() const { in pooling_width() function
225 return (input_width() + pooling_width() - 1) / pooling_width(); in output_width()
[all …]
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() argument
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 …]
Daverage-pooling-operator-tester.h100 (output_width() - 1) * stride_width() + pooling_width() - input_width(); in padding_left()
132 (output_width() - 1) * stride_width() + pooling_width() - input_width(); in padding_right()
194 …inline AveragePoolingOperatorTester& 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 AveragePoolingOperatorTester& pooling_width(uint32_t pooling_width) { in pooling_width() function
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
275 if (padded_input_width <= pooling_width()) { in output_width()
[all …]
Dmax-pooling-operator-tester.h193 inline MaxPoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width) { in pooling_size() argument
195 assert(pooling_width >= 1); in pooling_size()
197 this->pooling_width_ = pooling_width; in pooling_size()
211 inline MaxPoolingOperatorTester& pooling_width(uint32_t pooling_width) { in pooling_width() function
212 assert(pooling_width >= 1); in pooling_width()
213 this->pooling_width_ = pooling_width; in pooling_width()
217 inline uint32_t pooling_width() const { in pooling_width() function
296 return (pooling_width() - 1) * dilation_width() + 1; in dilated_pooling_width()
470 for (size_t px = 0; px < pooling_width(); px++) { in TestU8()
494 pooling_height(), pooling_width(), in TestU8()
[all …]
/external/XNNPACK/src/subgraph/
Dunpooling-2d.c23 uint32_t pooling_width, in xnn_define_unpooling_2d() argument
35 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_define_unpooling_2d()
40 xnn_node_type_to_string(xnn_node_type_unpooling_2d), pooling_width, pooling_height); in xnn_define_unpooling_2d()
83 node->params.pooling_2d.pooling_width = pooling_width; in xnn_define_unpooling_2d()
Dargmax-pooling-2d.c23 uint32_t pooling_width, in xnn_define_argmax_pooling_2d() argument
35 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_define_argmax_pooling_2d()
40 xnn_node_type_to_string(xnn_node_type_argmax_pooling_2d), pooling_width, pooling_height); in xnn_define_argmax_pooling_2d()
83 node->params.pooling_2d.pooling_width = pooling_width; in xnn_define_argmax_pooling_2d()
Daverage-pooling-2d.c23 uint32_t pooling_width, in xnn_define_average_pooling_2d() argument
38 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_define_average_pooling_2d()
43 xnn_node_type_to_string(xnn_node_type_average_pooling_2d), pooling_width, pooling_height); in xnn_define_average_pooling_2d()
120 node->params.pooling_2d.pooling_width = pooling_width; in xnn_define_average_pooling_2d()
Dmax-pooling-2d.c23 uint32_t pooling_width, in xnn_define_max_pooling_2d() argument
40 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_define_max_pooling_2d()
45 xnn_node_type_to_string(xnn_node_type_max_pooling_2d), pooling_width, pooling_height); in xnn_define_max_pooling_2d()
128 node->params.pooling_2d.pooling_width = pooling_width; in xnn_define_max_pooling_2d()
/external/XNNPACK/src/operators/
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()
Dmax-pooling-nhwc.c51 uint32_t pooling_width, in create_max_pooling2d_nhwc() argument
86 const uint32_t pooling_size = pooling_height * pooling_width; in create_max_pooling2d_nhwc()
92 pooling_width, pooling_height); in create_max_pooling2d_nhwc()
168 max_pooling_op->kernel_width = pooling_width; in create_max_pooling2d_nhwc()
260 const size_t pooling_width = max_pooling_op->kernel_width; in setup_max_pooling2d_nhwc() local
261 const size_t pooling_size = pooling_height * pooling_width; in setup_max_pooling2d_nhwc()
267 …max_pooling_op->dilation_width > 1 ? pooling_width : min(max_pooling_op->stride_width, pooling_wid… in setup_max_pooling2d_nhwc()
330 uint32_t pooling_width, in xnn_create_max_pooling2d_nhwc_u8() argument
353 pooling_height, pooling_width, in xnn_create_max_pooling2d_nhwc_u8()
369 uint32_t pooling_width, in xnn_create_max_pooling2d_nhwc_f32() argument
[all …]
Dunpooling-nhwc.c38 uint32_t pooling_width, in xnn_create_unpooling2d_nhwc_x32() argument
56 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_create_unpooling2d_nhwc_x32()
61 … xnn_operator_type_to_string(xnn_operator_type_unpooling_nhwc_x32), pooling_width, pooling_height); in xnn_create_unpooling2d_nhwc_x32()
111 unpooling_op->kernel_width = pooling_width; in xnn_create_unpooling2d_nhwc_x32()
191 const size_t pooling_width = unpooling_op->kernel_width; in xnn_setup_unpooling2d_nhwc_x32() local
192 const size_t pooling_size = pooling_height * pooling_width; in xnn_setup_unpooling2d_nhwc_x32()
Daverage-pooling-nhwc.c49 uint32_t pooling_width, in xnn_create_average_pooling2d_nhwc_qu8() argument
75 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_create_average_pooling2d_nhwc_qu8()
80 …xnn_operator_type_to_string(xnn_operator_type_average_pooling_nhwc_qu8), pooling_width, pooling_he… in xnn_create_average_pooling2d_nhwc_qu8()
171 pooling_size, pooling_width, pooling_height); in xnn_create_average_pooling2d_nhwc_qu8()
202 average_pooling_op->kernel_width = pooling_width; in xnn_create_average_pooling2d_nhwc_qu8()
245 uint32_t pooling_width, in xnn_create_average_pooling2d_nhwc_f32() argument
267 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_create_average_pooling2d_nhwc_f32()
272 …xnn_operator_type_to_string(xnn_operator_type_average_pooling_nhwc_f32), pooling_width, pooling_he… in xnn_create_average_pooling2d_nhwc_f32()
372 average_pooling_op->kernel_width = pooling_width; in xnn_create_average_pooling2d_nhwc_f32()
509 const size_t pooling_width = average_pooling_op->kernel_width; in setup_average_pooling2d() local
[all …]
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dpool_2d_tester.h91 inline Pool2DTester& PoolingWidth(int32_t pooling_width) { in PoolingWidth() argument
92 EXPECT_GT(pooling_width, 0); in PoolingWidth()
93 pooling_width_ = pooling_width; in PoolingWidth()
/external/XNNPACK/src/
Dindirection.c263 const size_t pooling_width = op->kernel_width; in xnn_indirection_init_maxpool2d() local
291 for (size_t pooling_x = 0; pooling_x < pooling_width; pooling_x++) { in xnn_indirection_init_maxpool2d()
316 for (size_t pooling_x = 0; pooling_x < pooling_width; pooling_x++) { in xnn_indirection_init_maxpool2d()
536 const size_t pooling_width = op->kernel_width; in xnn_indirection_init_unpool2d() local
545 for (size_t pooling_x = 0; pooling_x < pooling_width; pooling_x++) { in xnn_indirection_init_unpool2d()
546 …const size_t output_x = min(doz(input_x * pooling_width + pooling_x, output_padding_left), output_… in xnn_indirection_init_unpool2d()
547 …_buffer[(((image * input_height + input_y) * input_width + input_x) * pooling_width + pooling_x) *… in xnn_indirection_init_unpool2d()
Druntime.c146 node->params.pooling_2d.pooling_width, in xnn_create_runtime_v2()
169 node->params.pooling_2d.pooling_width, in xnn_create_runtime_v2()
559 node->params.pooling_2d.pooling_width, in xnn_create_runtime_v2()
826 node->params.pooling_2d.pooling_width, in xnn_create_runtime_v2()
/external/XNNPACK/include/
Dxnnpack.h439 uint32_t pooling_width,
531 uint32_t pooling_width,
566 uint32_t pooling_width,
596 uint32_t pooling_width,
1139 uint32_t pooling_width,
1162 uint32_t pooling_width,
1409 uint32_t pooling_width,
1784 uint32_t pooling_width,
1997 uint32_t pooling_width,
2229 uint32_t pooling_width,
/external/XNNPACK/src/xnnpack/
Dsubgraph.h178 uint32_t pooling_width; member