/aosp_15_r20/external/XNNPACK/test/ |
H A D | max-pooling-nhwc.cc | 25 .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 D | average-pooling-nhwc.cc | 42 .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 D | argmax-pooling-nhwc.cc | 38 .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 D | unpooling-nhwc.cc | 20 .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 D | argmax-pooling-operator-tester.h | 93 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 D | max-pooling-2d.cc | 45 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 D | unpooling-operator-tester.h | 143 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 D | average-pooling-2d.cc | 35 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 D | average-pooling-operator-tester.h | 101 (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 D | unpooling-2d.cc | 41 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 D | max-pooling-operator-tester.h | 194 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 D | argmax-pooling-2d.cc | 41 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 D | max-pooling-nhwc.c | 43 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 D | average-pooling-nhwc.c | 43 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 D | argmax-pooling-nhwc.c | 48 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 D | unpooling-nhwc.c | 30 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 D | op_max_dim.py | 65 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 D | op_avg_pooling2d.py | 47 pooling_height, pooling_width = cast(List, node.args[1]) 64 pooling_width=pooling_width,
|
/aosp_15_r20/external/XNNPACK/src/subgraph/ |
H A D | average-pooling-2d.c | 48 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 D | max-pooling-2d.c | 50 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 D | unpooling-2d.c | 50 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 D | argmax-pooling-2d.c | 51 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 D | average-pooling.c | 36 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 D | max-pooling.c | 41 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 D | indirection.c | 273 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 …]
|