Home
last modified time | relevance | path

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

/external/XNNPACK/test/
Dmax-pooling-nhwc.cc25 .pooling_width(pool_size) in TEST()
45 .pooling_width(pool_size) in TEST()
63 .pooling_width(pool_size) in TEST()
80 .pooling_width(pool_size) in TEST()
97 .pooling_width(1) in TEST()
117 .pooling_width(1) in TEST()
135 .pooling_width(1) in TEST()
152 .pooling_width(1) in TEST()
169 .pooling_width(1) in TEST()
178 .pooling_width(pool_size) in TEST()
[all …]
Daverage-pooling-nhwc.cc25 .pooling_width(pool_size) in TEST()
45 .pooling_width(pool_size) in TEST()
63 .pooling_width(pool_size) in TEST()
80 .pooling_width(1) in TEST()
100 .pooling_width(1) in TEST()
118 .pooling_width(1) in TEST()
135 .pooling_width(1) in TEST()
144 .pooling_width(pool_size) in TEST()
161 .pooling_width(1) in TEST()
170 .pooling_width(pool_size) in TEST()
[all …]
Dargmax-pooling-nhwc.cc38 .pooling_width(pool_size) in TEST()
58 .pooling_width(pool_size) in TEST()
76 .pooling_width(1) in TEST()
96 .pooling_width(1) in TEST()
114 .pooling_width(1) in TEST()
123 .pooling_width(pool_size) in TEST()
140 .pooling_width(1) in TEST()
149 .pooling_width(pool_size) in TEST()
166 .pooling_width(1) in TEST()
175 .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.h135 …inline ArgmaxPoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width) { in pooling_size() argument
137 assert(pooling_width >= 1); in pooling_size()
139 this->pooling_width_ = pooling_width; in pooling_size()
153 inline ArgmaxPoolingOperatorTester& pooling_width(uint32_t pooling_width) { in pooling_width() function
154 assert(pooling_width >= 1); in pooling_width()
155 this->pooling_width_ = pooling_width; in pooling_width()
159 inline uint32_t pooling_width() const { in pooling_width() function
170 return padded_input_width / pooling_width(); in output_width()
246 return padded_next_input_width / pooling_width(); in next_output_width()
310 …const size_t ix_top_left = std::max<size_t>(ox * pooling_width(), padding_left()) - padding_left(); in TestF32()
[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() 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 …]
Daverage-pooling-operator-tester.h146 …inline AveragePoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width)… in pooling_size() argument
148 assert(pooling_width >= 1); in pooling_size()
150 this->pooling_width_ = pooling_width; in pooling_size()
164 inline AveragePoolingOperatorTester& pooling_width(uint32_t pooling_width) { in pooling_width() argument
165 assert(pooling_width >= 1); in pooling_width()
166 this->pooling_width_ = pooling_width; in pooling_width()
170 inline uint32_t pooling_width() const { in pooling_width() function
220 if (padded_input_width <= pooling_width()) { in output_width()
223 return (padded_input_width - pooling_width()) / stride_width() + 1; in output_width()
304 if (padded_next_input_width <= pooling_width()) { in next_output_width()
[all …]
Dmax-pooling-operator-tester.h146 inline MaxPoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width) { in pooling_size() argument
148 assert(pooling_width >= 1); in pooling_size()
150 this->pooling_width_ = pooling_width; in pooling_size()
164 inline MaxPoolingOperatorTester& pooling_width(uint32_t pooling_width) { in pooling_width() function
165 assert(pooling_width >= 1); in pooling_width()
166 this->pooling_width_ = pooling_width; in pooling_width()
170 inline uint32_t pooling_width() const { in pooling_width() function
249 return (pooling_width() - 1) * dilation_width() + 1; in dilated_pooling_width()
415 for (size_t px = 0; px < pooling_width(); px++) { in TestU8()
438 pooling_height(), pooling_width(), in TestU8()
[all …]
/external/XNNPACK/src/
Dmax-pooling-nhwc.c44 uint32_t pooling_width, in xnn_create_max_pooling2d_nhwc_u8() argument
67 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_create_max_pooling2d_nhwc_u8()
72 pooling_width, pooling_height); in xnn_create_max_pooling2d_nhwc_u8()
143 max_pooling_op->kernel_width = pooling_width; in xnn_create_max_pooling2d_nhwc_u8()
173 uint32_t pooling_width, in xnn_create_max_pooling2d_nhwc_f32() argument
196 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_create_max_pooling2d_nhwc_f32()
201 pooling_width, pooling_height); in xnn_create_max_pooling2d_nhwc_f32()
284 max_pooling_op->kernel_width = pooling_width; in xnn_create_max_pooling2d_nhwc_f32()
357 const size_t pooling_width = max_pooling_op->kernel_width; in setup_max_pooling2d() local
358 const size_t pooling_size = pooling_height * pooling_width; in setup_max_pooling2d()
[all …]
Dargmax-pooling-nhwc.c48 uint32_t pooling_width, in xnn_create_argmax_pooling2d_nhwc_f32() argument
67 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_create_argmax_pooling2d_nhwc_f32()
72 pooling_width, pooling_height); in xnn_create_argmax_pooling2d_nhwc_f32()
143 argmax_pooling_op->kernel_width = pooling_width; in xnn_create_argmax_pooling2d_nhwc_f32()
145 argmax_pooling_op->stride_width = pooling_width; in xnn_create_argmax_pooling2d_nhwc_f32()
213 const size_t pooling_width = argmax_pooling_op->kernel_width; in xnn_setup_argmax_pooling2d_nhwc_f32() local
214 const size_t pooling_size = pooling_height * pooling_width; in xnn_setup_argmax_pooling2d_nhwc_f32()
220 const size_t step_width = pooling_width; in xnn_setup_argmax_pooling2d_nhwc_f32()
Daverage-pooling-nhwc.c42 uint32_t pooling_width, in xnn_create_average_pooling2d_nhwc_q8() argument
67 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_create_average_pooling2d_nhwc_q8()
72 pooling_width, pooling_height); in xnn_create_average_pooling2d_nhwc_q8()
152 pooling_size, pooling_width, pooling_height); in xnn_create_average_pooling2d_nhwc_q8()
184 average_pooling_op->kernel_width = pooling_width; in xnn_create_average_pooling2d_nhwc_q8()
218 uint32_t pooling_width, in xnn_create_average_pooling2d_nhwc_f32() argument
239 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_create_average_pooling2d_nhwc_f32()
244 pooling_width, pooling_height); in xnn_create_average_pooling2d_nhwc_f32()
331 average_pooling_op->kernel_width = pooling_width; in xnn_create_average_pooling2d_nhwc_f32()
423 const size_t pooling_width = average_pooling_op->kernel_width; in xnn_setup_average_pooling2d_nhwc_q8() local
[all …]
Dunpooling-nhwc.c38 uint32_t pooling_width, in xnn_create_unpooling2d_nhwc_x32() argument
55 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_create_unpooling2d_nhwc_x32()
60 pooling_width, pooling_height); in xnn_create_unpooling2d_nhwc_x32()
107 unpooling_op->kernel_width = pooling_width; in xnn_create_unpooling2d_nhwc_x32()
185 const size_t pooling_width = unpooling_op->kernel_width; in xnn_setup_unpooling2d_nhwc_x32() local
186 const size_t pooling_size = pooling_height * pooling_width; in xnn_setup_unpooling2d_nhwc_x32()
Dindirection.c267 const size_t pooling_width = op->kernel_width; in xnn_indirection_init_maxpool2d() local
280 for (size_t pooling_x = 0; pooling_x < pooling_width; pooling_x++) { in xnn_indirection_init_maxpool2d()
400 const size_t pooling_width = op->kernel_width; in xnn_indirection_init_unpool2d() local
409 for (size_t pooling_x = 0; pooling_x < pooling_width; pooling_x++) { in xnn_indirection_init_unpool2d()
410 …const size_t output_x = min(doz(input_x * pooling_width + pooling_x, output_padding_left), output_… in xnn_indirection_init_unpool2d()
411 …_buffer[(((image * input_height + input_y) * input_width + input_x) * pooling_width + pooling_x) *… in xnn_indirection_init_unpool2d()
/external/XNNPACK/include/
Dxnnpack.h530 uint32_t pooling_width,
555 uint32_t pooling_width,
730 uint32_t pooling_width,
972 uint32_t pooling_width,
1023 uint32_t pooling_width,
1255 uint32_t pooling_width,