| /aosp_15_r20/external/XNNPACK/test/ |
| H A D | average-pooling-nhwc.cc | 36 const std::pair<size_t, size_t> pooling_size = SmallPoolSize(xnn_params.qu8.avgpool.primary_tile); in TEST() local
57 const std::pair<size_t, size_t> pooling_size = SmallPoolSize(xnn_params.qu8.avgpool.primary_tile); in TEST() local
90 const std::pair<size_t, size_t> pooling_size = SmallPoolSize(xnn_params.qu8.avgpool.primary_tile); in TEST() local
123 const std::pair<size_t, size_t> pooling_size = SmallPoolSize(xnn_params.qu8.avgpool.primary_tile); in TEST() local
156 const std::pair<size_t, size_t> pooling_size = SmallPoolSize(xnn_params.qu8.avgpool.primary_tile); in TEST() local
209 const std::pair<size_t, size_t> pooling_size = SmallPoolSize(xnn_params.qu8.avgpool.primary_tile); in TEST() local
232 const std::pair<size_t, size_t> pooling_size = SmallPoolSize(xnn_params.qu8.avgpool.primary_tile); in TEST() local
255 const std::pair<size_t, size_t> pooling_size = SmallPoolSize(xnn_params.qu8.avgpool.primary_tile); in TEST() local
280 const std::pair<size_t, size_t> pooling_size = SmallPoolSize(xnn_params.qu8.avgpool.primary_tile); in TEST() local
305 const std::pair<size_t, size_t> pooling_size = SmallPoolSize(xnn_params.qu8.avgpool.primary_tile); in TEST() local
[all …]
|
| H A D | unpooling-operator-tester.h | 136 inline UnpoolingOperatorTester& pooling_size(uint32_t pooling_size) { in pooling_size() function
143 inline UnpoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width) { in pooling_size() function
|
| H A D | argmax-pooling-operator-tester.h | 178 inline ArgmaxPoolingOperatorTester& pooling_size(uint32_t pooling_size) { in pooling_size() argument
185 …inline ArgmaxPoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width) { in pooling_size() function
|
| H A D | average-pooling-operator-tester.h | 188 inline AveragePoolingOperatorTester& pooling_size(uint32_t pooling_size) { in pooling_size() function
195 …inline AveragePoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width)… in pooling_size() function
|
| H A D | max-pooling-operator-tester.h | 187 inline MaxPoolingOperatorTester& pooling_size(uint32_t pooling_size) { in pooling_size() argument
194 inline MaxPoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width) { in pooling_size() function
|
| /aosp_15_r20/external/XNNPACK/src/operators/ |
| H A D | average-pooling-nhwc.c | 69 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_create_average_pooling2d_nhwc_qu8() local
282 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_create_average_pooling2d_nhwc_f16() local
462 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_create_average_pooling2d_nhwc_f32() local
721 const size_t pooling_size = pooling_height * pooling_width; in setup_average_pooling2d() local
864 const size_t pooling_size = average_pooling_op->kernel_height * average_pooling_op->kernel_width; in xnn_setup_average_pooling2d_nhwc_qu8() local
|
| H A D | argmax-pooling-nhwc.c | 33 size_t pooling_size, in select_ukernel()
66 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_create_argmax_pooling2d_nhwc_f32() local
218 const size_t pooling_size = pooling_height * pooling_width; in xnn_setup_argmax_pooling2d_nhwc_f32() local
|
| H A D | unpooling-nhwc.c | 48 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_create_unpooling2d_nhwc_x32() local
185 const size_t pooling_size = pooling_height * pooling_width; in xnn_setup_unpooling2d_nhwc_x32() local
|
| H A D | max-pooling-nhwc.c | 78 const uint32_t pooling_size = pooling_height * pooling_width; in create_max_pooling2d_nhwc() local
274 const size_t pooling_size = pooling_height * pooling_width; in setup_max_pooling2d_nhwc() local
|
| /aosp_15_r20/external/XNNPACK/bench/ |
| H A D | average-pooling.cc | 35 const size_t pooling_size = state.range(3); in xnnpack_average_pooling_qu8() local
114 const size_t pooling_size = state.range(3); in xnnpack_average_pooling_f32() local
191 const size_t pooling_size = state.range(3); in tflite_average_pooling_f32() local
|
| H A D | max-pooling.cc | 27 const size_t pooling_size = state.range(3); in max_pooling_u8() local
104 const size_t pooling_size = state.range(3); in max_pooling_f32() local
|
| /aosp_15_r20/external/XNNPACK/src/xnnpack/ |
| H A D | compute.h | 557 size_t pooling_size; member
585 size_t pooling_size; member
610 size_t pooling_size; member
641 size_t pooling_size; member
680 size_t pooling_size; member
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/ |
| H A D | max-pooling.c | 62 const uint32_t pooling_size = pooling_height * pooling_width; in pytorch_qnnp_create_max_pooling2d_nhwc_u8() local
|
| H A D | average-pooling.c | 59 const uint32_t pooling_size = pooling_height * pooling_width; in pytorch_qnnp_create_average_pooling2d_nhwc_q8() local
|
| H A D | operator-run.c | 535 size_t pooling_size; member
572 size_t pooling_size; member
1338 const size_t pooling_size = pooling_height * pooling_width; in pytorch_qnnp_run_operator() local
1404 const size_t pooling_size = pooling_height * pooling_width; in pytorch_qnnp_run_operator() local
|
| /aosp_15_r20/external/XNNPACK/src/subgraph/ |
| H A D | unpooling-2d.c | 124 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_define_unpooling_2d() local
|
| H A D | average-pooling-2d.c | 156 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_define_average_pooling_2d() local
|
| H A D | argmax-pooling-2d.c | 125 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_define_argmax_pooling_2d() local
|
| H A D | max-pooling-2d.c | 238 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_define_max_pooling_2d() local
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | lowering.py | 3752 def pooling_size(x, i, kernel_size, stride, padding, ceil_mode): function
|