| /external/XNNPACK/test/ |
| D | average-pooling-nhwc.cc | 36 const std::pair<size_t, size_t> pooling_size = SmallPoolSize(xnn_params.qu8.avgpool.mr); in TEST() local
39 .input_height(pooling_size.first + 3) in TEST()
40 .input_width(pooling_size.second + 2) in TEST()
41 .pooling_height(pooling_size.first) in TEST()
42 .pooling_width(pooling_size.second) in TEST()
46 .input_height(pooling_size.second + 3) in TEST()
47 .input_width(pooling_size.first + 2) in TEST()
48 .pooling_height(pooling_size.second) in TEST()
49 .pooling_width(pooling_size.first) in TEST()
57 const std::pair<size_t, size_t> pooling_size = SmallPoolSize(xnn_params.qu8.avgpool.mr); in TEST() local
[all …]
|
| D | unpooling-nhwc.cc | 14 for (size_t pooling_size : std::vector<size_t>{{2, 3, 5, 7}}) { in TEST()
20 .pooling_width(pooling_size) in TEST()
30 for (size_t pooling_size : std::vector<size_t>{{3, 5, 7}}) { in TEST()
40 .pooling_width(pooling_size) in TEST()
52 for (size_t pooling_size : std::vector<size_t>{{2, 3, 5, 7}}) { in TEST()
57 .pooling_height(pooling_size) in TEST()
68 for (size_t pooling_size : std::vector<size_t>{{3, 5, 7}}) { in TEST()
77 .pooling_height(pooling_size) in TEST()
90 for (size_t pooling_size : std::vector<size_t>{{2, 3, 5}}) { in TEST()
95 .pooling_height(pooling_size) in TEST()
[all …]
|
| D | unpooling-operator-tester.h | 136 inline UnpoolingOperatorTester& pooling_size(uint32_t pooling_size) { in pooling_size() function
137 assert(pooling_size >= 1); in pooling_size()
138 this->pooling_height_ = pooling_size; in pooling_size()
139 this->pooling_width_ = pooling_size; in pooling_size()
143 inline UnpoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width) { in pooling_size() function
|
| D | argmax-pooling-operator-tester.h | 179 inline ArgmaxPoolingOperatorTester& pooling_size(uint32_t pooling_size) { in pooling_size() argument
180 assert(pooling_size >= 1); in pooling_size()
181 this->pooling_height_ = pooling_size; in pooling_size()
182 this->pooling_width_ = pooling_size; in pooling_size()
186 …inline ArgmaxPoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width) { in pooling_size() function
|
| D | average-pooling-operator-tester.h | 187 inline AveragePoolingOperatorTester& pooling_size(uint32_t pooling_size) { in pooling_size() function
188 assert(pooling_size >= 1); in pooling_size()
189 this->pooling_height_ = pooling_size; in pooling_size()
190 this->pooling_width_ = pooling_size; in pooling_size()
194 …inline AveragePoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width)… in pooling_size() function
|
| D | max-pooling-operator-tester.h | 186 inline MaxPoolingOperatorTester& pooling_size(uint32_t pooling_size) { in pooling_size() argument
187 assert(pooling_size >= 1); in pooling_size()
188 this->pooling_height_ = pooling_size; in pooling_size()
189 this->pooling_width_ = pooling_size; in pooling_size()
193 inline MaxPoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width) { in pooling_size() function
|
| /external/XNNPACK/src/operators/ |
| D | argmax-pooling-nhwc.c | 33 size_t pooling_size, in select_ukernel() argument
36 while (ukernel->qr == 0 && ukernel->mr < pooling_size) { in select_ukernel()
66 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_create_argmax_pooling2d_nhwc_f32() local
67 if (pooling_size == 0) { in xnn_create_argmax_pooling2d_nhwc_f32()
75 if (pooling_size == 1) { in xnn_create_argmax_pooling2d_nhwc_f32()
218 const size_t pooling_size = pooling_height * pooling_width; in xnn_setup_argmax_pooling2d_nhwc_f32() local
221 …const struct argmaxpool_parameters* ukernel = select_ukernel(pooling_size, xnn_params.f32.argmaxpo… in xnn_setup_argmax_pooling2d_nhwc_f32()
225 const size_t step_height = pooling_size + (output_width - 1) * step_width * pooling_height; in xnn_setup_argmax_pooling2d_nhwc_f32()
258 const size_t multipass_adjustment = qr == 0 ? 0 : round_up(pooling_size - mr, qr) + mr - qr; in xnn_setup_argmax_pooling2d_nhwc_f32()
271 .pooling_size = pooling_size, in xnn_setup_argmax_pooling2d_nhwc_f32()
[all …]
|
| D | average-pooling-nhwc.c | 75 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_create_average_pooling2d_nhwc_qu8() local
76 if (pooling_size == 0) { in xnn_create_average_pooling2d_nhwc_qu8()
84 if (pooling_size == 1) { in xnn_create_average_pooling2d_nhwc_qu8()
166 if (pooling_size >= 16777216) { in xnn_create_average_pooling2d_nhwc_qu8()
171 pooling_size, pooling_width, pooling_height); in xnn_create_average_pooling2d_nhwc_qu8()
220 …round_up(doz(pooling_size, xnn_params.qu8.avgpool.mr), xnn_params.qu8.avgpool.qr) + xnn_params.qu8… in xnn_create_average_pooling2d_nhwc_qu8()
224 input_scale / (output_scale * (float) pooling_size), in xnn_create_average_pooling2d_nhwc_qu8()
267 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_create_average_pooling2d_nhwc_f32() local
268 if (pooling_size == 0) { in xnn_create_average_pooling2d_nhwc_f32()
276 if (pooling_size == 1) { in xnn_create_average_pooling2d_nhwc_f32()
[all …]
|
| D | unpooling-nhwc.c | 56 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_create_unpooling2d_nhwc_x32() local
57 if (pooling_size == 0) { in xnn_create_unpooling2d_nhwc_x32()
65 if (pooling_size == 1) { in xnn_create_unpooling2d_nhwc_x32()
192 const size_t pooling_size = pooling_height * pooling_width; in xnn_setup_unpooling2d_nhwc_x32() local
194 …indirection_buffer_size = sizeof(void*) * (batch_size * input_height * input_width * pooling_size); in xnn_setup_unpooling2d_nhwc_x32()
216 .indirect_output_height_stride = input_width * pooling_size * sizeof(void*), in xnn_setup_unpooling2d_nhwc_x32()
217 .indirect_output_width_stride = pooling_size * sizeof(void*), in xnn_setup_unpooling2d_nhwc_x32()
218 .pooling_size = pooling_size, in xnn_setup_unpooling2d_nhwc_x32()
|
| D | max-pooling-nhwc.c | 86 const uint32_t pooling_size = pooling_height * pooling_width; in create_max_pooling2d_nhwc() local
87 if (pooling_size == 0) { in create_max_pooling2d_nhwc()
96 if (pooling_size == 1) { in create_max_pooling2d_nhwc()
261 const size_t pooling_size = pooling_height * pooling_width; in setup_max_pooling2d_nhwc() local
268 const size_t step_height = pooling_size + (output_width - 1) * step_width * pooling_height; in setup_max_pooling2d_nhwc()
296 const size_t multipass_adjustment = round_up(doz(pooling_size, mr), qr) + mr; in setup_max_pooling2d_nhwc()
307 .pooling_size = pooling_size, in setup_max_pooling2d_nhwc()
|
| /external/XNNPACK/bench/ |
| D | max-pooling.cc | 27 const size_t pooling_size = state.range(3); in max_pooling_u8() local
36 const size_t output_height = (2 * padding_size + input_height - pooling_size) / stride + 1; in max_pooling_u8()
37 const size_t output_width = (2 * padding_size + input_width - pooling_size) / stride + 1; in max_pooling_u8()
53 pooling_size, pooling_size, in max_pooling_u8()
104 const size_t pooling_size = state.range(3); in max_pooling_f32() local
113 const size_t output_height = (2 * padding_size + input_height - pooling_size) / stride + 1; in max_pooling_f32()
114 const size_t output_width = (2 * padding_size + input_width - pooling_size) / stride + 1; in max_pooling_f32()
130 pooling_size, pooling_size, in max_pooling_f32()
|
| D | average-pooling.cc | 35 const size_t pooling_size = state.range(3); in xnnpack_average_pooling_qu8() local
44 const size_t output_height = (2 * padding_size + input_height - pooling_size) / stride + 1; in xnnpack_average_pooling_qu8()
45 const size_t output_width = (2 * padding_size + input_width - pooling_size) / stride + 1; in xnnpack_average_pooling_qu8()
61 pooling_size, pooling_size, in xnnpack_average_pooling_qu8()
114 const size_t pooling_size = state.range(3); in xnnpack_average_pooling_f32() local
123 const size_t output_height = (2 * padding_size + input_height - pooling_size) / stride + 1; in xnnpack_average_pooling_f32()
124 const size_t output_width = (2 * padding_size + input_width - pooling_size) / stride + 1; in xnnpack_average_pooling_f32()
140 pooling_size, pooling_size, in xnnpack_average_pooling_f32()
191 const size_t pooling_size = state.range(3); in tflite_average_pooling_f32() local
201 if (2 * padding_size == (pooling_size - 1)) { in tflite_average_pooling_f32()
[all …]
|
| /external/XNNPACK/src/subgraph/ |
| D | unpooling-2d.c | 35 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_define_unpooling_2d() local
36 if (pooling_size == 0) { in xnn_define_unpooling_2d()
44 if (pooling_size == 1) { in xnn_define_unpooling_2d()
|
| D | argmax-pooling-2d.c | 35 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_define_argmax_pooling_2d() local
36 if (pooling_size == 0) { in xnn_define_argmax_pooling_2d()
44 if (pooling_size == 1) { in xnn_define_argmax_pooling_2d()
|
| D | average-pooling-2d.c | 38 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_define_average_pooling_2d() local
39 if (pooling_size == 0) { in xnn_define_average_pooling_2d()
47 if (pooling_size == 1) { in xnn_define_average_pooling_2d()
|
| D | max-pooling-2d.c | 40 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_define_max_pooling_2d() local
41 if (pooling_size == 0) { in xnn_define_max_pooling_2d()
49 if (pooling_size == 1) { in xnn_define_max_pooling_2d()
|
| /external/XNNPACK/src/xnnpack/ |
| D | compute.h | 498 size_t pooling_size; member
526 size_t pooling_size; member
551 size_t pooling_size; member
582 size_t pooling_size; member
620 size_t pooling_size; member
|
| /external/XNNPACK/src/ |
| D | operator-run.c | 473 context->output_width, context->pooling_size, context->channels, in xnn_compute_argmax_pooling_unipass()
495 context->output_width, context->pooling_size, context->channels, in xnn_compute_argmax_pooling_multipass()
512 context->output_width, context->pooling_size, context->channels, in xnn_compute_max_pooling()
532 context->pooling_size, in xnn_compute_unpooling()
550 context->output_width, context->pooling_size, context->channels, in xnn_compute_average_pooling_unipass()
571 context->output_width, context->pooling_size, context->channels, in xnn_compute_average_pooling_multipass()
591 context->output_width, context->pooling_size, context->channels, in xnn_compute_pixelwise_average_pooling_unipass()
613 context->output_width, context->pooling_size, context->channels, in xnn_compute_pixelwise_average_pooling_multipass()
|