| /external/XNNPACK/test/ |
| D | max-pooling-nhwc.cc | 24 .pooling_height(1) in TEST()
44 .pooling_height(1) in TEST()
64 .pooling_height(1) in TEST()
81 .pooling_height(1) in TEST()
100 .pooling_height(1) in TEST()
117 .pooling_height(pool_size) in TEST()
137 .pooling_height(pool_size) in TEST()
157 .pooling_height(pool_size) in TEST()
174 .pooling_height(pool_size) in TEST()
193 .pooling_height(pool_size) in TEST()
[all …]
|
| D | average-pooling-nhwc.cc | 41 .pooling_height(pooling_size.first) in TEST()
48 .pooling_height(pooling_size.second) in TEST()
68 .pooling_height(pooling_size.first) in TEST()
77 .pooling_height(pooling_size.second) in TEST()
100 .pooling_height(pooling_size.first) in TEST()
110 .pooling_height(pooling_size.second) in TEST()
133 .pooling_height(pooling_size.first) in TEST()
143 .pooling_height(pooling_size.second) in TEST()
164 .pooling_height(pooling_size.first) in TEST()
175 .pooling_height(pooling_size.first) in TEST()
[all …]
|
| D | argmax-pooling-nhwc.cc | 37 .pooling_height(1) in TEST()
57 .pooling_height(1) in TEST()
77 .pooling_height(1) in TEST()
94 .pooling_height(pool_size) in TEST()
114 .pooling_height(pool_size) in TEST()
134 .pooling_height(pool_size) in TEST()
151 .pooling_height(pool_size) in TEST()
160 .pooling_height(1) in TEST()
177 .pooling_height(pool_size) in TEST()
186 .pooling_height(1) in TEST()
[all …]
|
| D | unpooling-nhwc.cc | 19 .pooling_height(1) in TEST()
39 .pooling_height(1) in TEST()
57 .pooling_height(pooling_size) in TEST()
77 .pooling_height(pooling_size) in TEST()
95 .pooling_height(pooling_size) in TEST()
118 .pooling_height(3) in TEST()
138 .pooling_height(pooling_size) in TEST()
153 .pooling_height(3) in TEST()
168 .pooling_height(3) in TEST()
187 .pooling_height(3) in TEST()
[all …]
|
| D | argmax-pooling-operator-tester.h | 79 const uint32_t total_padding_height = output_height() * pooling_height() - input_height(); in padding_top()
109 const uint32_t total_padding_height = output_height() * pooling_height() - input_height(); in padding_bottom()
186 …inline ArgmaxPoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width) { in pooling_size() argument
187 assert(pooling_height >= 1); in pooling_size()
189 this->pooling_height_ = pooling_height; in pooling_size()
194 inline ArgmaxPoolingOperatorTester& pooling_height(uint32_t pooling_height) { in pooling_height() argument
195 assert(pooling_height >= 1); in pooling_height()
196 this->pooling_height_ = pooling_height; in pooling_height()
200 inline uint32_t pooling_height() const { in pooling_height() function
216 return (input_height() + pooling_height() - 1) / pooling_height(); in output_height()
[all …]
|
| D | unpooling-operator-tester.h | 143 inline UnpoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width) { in pooling_size() argument
144 assert(pooling_height >= 1); in pooling_size()
146 this->pooling_height_ = pooling_height; in pooling_size()
151 inline UnpoolingOperatorTester& pooling_height(uint32_t pooling_height) { in pooling_height() argument
152 assert(pooling_height >= 1); in pooling_height()
153 this->pooling_height_ = pooling_height; in pooling_height()
157 inline uint32_t pooling_height() const { in pooling_height() function
173 return std::max<size_t>(input_height() * pooling_height(), padding_height) - padding_height; in output_height()
249 … return std::max<size_t>(next_input_height() * pooling_height(), padding_height) - padding_height; in next_output_height()
284 …auto idx_rng = std::bind(std::uniform_int_distribution<uint32_t>(0, pooling_height() * pooling_wid… in TestX32()
[all …]
|
| D | average-pooling-operator-tester.h | 84 (output_height() - 1) * stride_height() + pooling_height() - input_height(); in padding_top()
116 (output_height() - 1) * stride_height() + pooling_height() - input_height(); in padding_bottom()
194 …inline AveragePoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width)… in pooling_size() argument
195 assert(pooling_height >= 1); in pooling_size()
197 this->pooling_height_ = pooling_height; in pooling_size()
202 inline AveragePoolingOperatorTester& pooling_height(uint32_t pooling_height) { in pooling_height() function
203 assert(pooling_height >= 1); in pooling_height()
204 this->pooling_height_ = pooling_height; in pooling_height()
208 inline uint32_t pooling_height() const { in pooling_height() function
262 if (padded_input_height <= pooling_height()) { in output_height()
[all …]
|
| D | max-pooling-operator-tester.h | 193 inline MaxPoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width) { in pooling_size() argument
194 assert(pooling_height >= 1); in pooling_size()
196 this->pooling_height_ = pooling_height; in pooling_size()
201 inline MaxPoolingOperatorTester& pooling_height(uint32_t pooling_height) { in pooling_height() argument
202 assert(pooling_height >= 1); in pooling_height()
203 this->pooling_height_ = pooling_height; in pooling_height()
207 inline uint32_t pooling_height() const { in pooling_height() function
292 return (pooling_height() - 1) * dilation_height() + 1; in dilated_pooling_height()
468 for (size_t py = 0; py < pooling_height(); py++) { in TestU8()
494 pooling_height(), pooling_width(), in TestU8()
[all …]
|
| /external/XNNPACK/src/operators/ |
| D | argmax-pooling-nhwc.c | 47 uint32_t pooling_height, 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 …operator_type_to_string(xnn_operator_type_argmax_pooling_nhwc_f32), pooling_width, pooling_height); in xnn_create_argmax_pooling2d_nhwc_f32()
132 argmax_pooling_op->kernel_height = pooling_height; in xnn_create_argmax_pooling2d_nhwc_f32()
134 argmax_pooling_op->stride_height = pooling_height; in xnn_create_argmax_pooling2d_nhwc_f32()
196 const size_t pooling_height = argmax_pooling_op->kernel_height; in xnn_setup_argmax_pooling2d_nhwc_f32() local
200 argmax_pooling_op->output_height = divide_round_up(input_height, pooling_height); in xnn_setup_argmax_pooling2d_nhwc_f32()
203 … const uint32_t padding_height = argmax_pooling_op->output_height * pooling_height - input_height; in xnn_setup_argmax_pooling2d_nhwc_f32()
218 const size_t pooling_size = pooling_height * pooling_width; 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()
[all …]
|
| D | max-pooling-nhwc.c | 50 uint32_t pooling_height, 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()
167 max_pooling_op->kernel_height = pooling_height; in create_max_pooling2d_nhwc()
259 const size_t pooling_height = max_pooling_op->kernel_height; in setup_max_pooling2d_nhwc() local
261 const size_t pooling_size = pooling_height * pooling_width; in setup_max_pooling2d_nhwc()
268 const size_t step_height = pooling_size + (output_width - 1) * step_width * pooling_height; in setup_max_pooling2d_nhwc()
309 .input_increment = (pooling_height * step_width - multipass_adjustment) * sizeof(void*), in setup_max_pooling2d_nhwc()
329 uint32_t pooling_height, in xnn_create_max_pooling2d_nhwc_u8() argument
353 pooling_height, pooling_width, in xnn_create_max_pooling2d_nhwc_u8()
[all …]
|
| D | unpooling-nhwc.c | 37 uint32_t pooling_height, 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()
110 unpooling_op->kernel_height = pooling_height; in xnn_create_unpooling2d_nhwc_x32()
190 const size_t pooling_height = unpooling_op->kernel_height; in xnn_setup_unpooling2d_nhwc_x32() local
192 const size_t pooling_size = pooling_height * pooling_width; in xnn_setup_unpooling2d_nhwc_x32()
|
| D | average-pooling-nhwc.c | 48 uint32_t pooling_height, 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 …perator_type_to_string(xnn_operator_type_average_pooling_nhwc_qu8), pooling_width, pooling_height); in xnn_create_average_pooling2d_nhwc_qu8()
171 pooling_size, pooling_width, pooling_height); in xnn_create_average_pooling2d_nhwc_qu8()
201 average_pooling_op->kernel_height = pooling_height; in xnn_create_average_pooling2d_nhwc_qu8()
244 uint32_t pooling_height, 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 …perator_type_to_string(xnn_operator_type_average_pooling_nhwc_f32), pooling_width, pooling_height); in xnn_create_average_pooling2d_nhwc_f32()
371 average_pooling_op->kernel_height = pooling_height; in xnn_create_average_pooling2d_nhwc_f32()
508 const size_t pooling_height = average_pooling_op->kernel_height; in setup_average_pooling2d() local
[all …]
|
| /external/XNNPACK/src/subgraph/ |
| D | unpooling-2d.c | 22 uint32_t pooling_height, 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()
82 node->params.pooling_2d.pooling_height = pooling_height; in xnn_define_unpooling_2d()
|
| D | argmax-pooling-2d.c | 22 uint32_t pooling_height, 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()
82 node->params.pooling_2d.pooling_height = pooling_height; in xnn_define_argmax_pooling_2d()
|
| D | average-pooling-2d.c | 22 uint32_t pooling_height, 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()
119 node->params.pooling_2d.pooling_height = pooling_height; in xnn_define_average_pooling_2d()
|
| D | max-pooling-2d.c | 22 uint32_t pooling_height, 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()
127 node->params.pooling_2d.pooling_height = pooling_height; in xnn_define_max_pooling_2d()
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | pool_2d_tester.h | 83 inline Pool2DTester& PoolingHeight(int32_t pooling_height) { in PoolingHeight() argument
84 EXPECT_GT(pooling_height, 0); in PoolingHeight()
85 pooling_height_ = pooling_height; in PoolingHeight()
|
| /external/XNNPACK/src/ |
| D | indirection.c | 262 const size_t pooling_height = op->kernel_height; in xnn_indirection_init_maxpool2d() local
278 for (size_t pooling_y = 0; pooling_y < pooling_height; pooling_y++) { in xnn_indirection_init_maxpool2d()
303 …dex = output_y * step_height + output_x * step_width * pooling_height + pooling_x * pooling_height… in xnn_indirection_init_maxpool2d()
313 for (size_t pooling_y = 0; pooling_y < pooling_height; pooling_y++) { in xnn_indirection_init_maxpool2d()
318 …dex = output_y * step_height + output_x * step_width * pooling_height + pooling_x * pooling_height… in xnn_indirection_init_maxpool2d()
535 const size_t pooling_height = op->kernel_height; in xnn_indirection_init_unpool2d() local
542 for (size_t pooling_y = 0; pooling_y < pooling_height; pooling_y++) { in xnn_indirection_init_unpool2d()
543 …const size_t output_y = min(doz(input_y * pooling_height + pooling_y, output_padding_top), output_… in xnn_indirection_init_unpool2d()
547 …ht + input_y) * input_width + input_x) * pooling_width + pooling_x) * pooling_height + pooling_y] = in xnn_indirection_init_unpool2d()
|
| D | runtime.c | 145 node->params.pooling_2d.pooling_height, in xnn_create_runtime_v2()
168 node->params.pooling_2d.pooling_height, in xnn_create_runtime_v2()
558 node->params.pooling_2d.pooling_height, in xnn_create_runtime_v2()
825 node->params.pooling_2d.pooling_height, in xnn_create_runtime_v2()
|
| /external/XNNPACK/include/ |
| D | xnnpack.h | 438 uint32_t pooling_height,
530 uint32_t pooling_height,
565 uint32_t pooling_height,
595 uint32_t pooling_height,
1138 uint32_t pooling_height,
1161 uint32_t pooling_height,
1408 uint32_t pooling_height,
1783 uint32_t pooling_height,
1996 uint32_t pooling_height,
2228 uint32_t pooling_height,
|
| /external/XNNPACK/src/xnnpack/ |
| D | subgraph.h | 177 uint32_t pooling_height; member
|