| /external/XNNPACK/test/ |
| D | f32-argmaxpool.cc | 24 .pooling_elements(4) in TEST()
33 .pooling_elements(4) in TEST()
42 for (size_t pooling_elements = 2; pooling_elements < 4; pooling_elements++) { in TEST() local
44 .pooling_elements(pooling_elements) in TEST()
53 for (size_t pooling_elements = 2; pooling_elements < 4; pooling_elements++) { in TEST() local
55 .pooling_elements(pooling_elements) in TEST()
67 .pooling_elements(4) in TEST()
78 .pooling_elements(4) in TEST()
88 for (size_t pooling_elements = 2; pooling_elements < 4; pooling_elements++) { in TEST() local
91 .pooling_elements(pooling_elements) in TEST()
[all …]
|
| D | f32-avgpool-minmax.cc | 28 .pooling_elements(17) in TEST()
37 .pooling_elements(17) in TEST()
48 .pooling_elements(17) in TEST()
60 .pooling_elements(17) in TEST()
70 .pooling_elements(17) in TEST()
79 for (size_t pooling_elements = 10; pooling_elements < 17; pooling_elements++) { in TEST() local
81 .pooling_elements(pooling_elements) in TEST()
90 for (size_t pooling_elements = 10; pooling_elements < 17; pooling_elements++) { in TEST() local
92 .pooling_elements(pooling_elements) in TEST()
102 for (size_t pooling_elements = 10; pooling_elements < 17; pooling_elements++) { in TEST() local
[all …]
|
| D | f32-pavgpool-minmax.cc | 28 .pooling_elements(17) in TEST()
37 .pooling_elements(17) in TEST()
48 .pooling_elements(17) in TEST()
60 .pooling_elements(17) in TEST()
70 .pooling_elements(17) in TEST()
79 for (size_t pooling_elements = 10; pooling_elements < 17; pooling_elements++) { in TEST() local
81 .pooling_elements(pooling_elements) in TEST()
90 for (size_t pooling_elements = 10; pooling_elements < 17; pooling_elements++) { in TEST() local
92 .pooling_elements(pooling_elements) in TEST()
102 for (size_t pooling_elements = 10; pooling_elements < 17; pooling_elements++) { in TEST() local
[all …]
|
| D | f32-maxpool-minmax.cc | 27 .pooling_elements(9) in TEST()
36 .pooling_elements(9) in TEST()
46 .pooling_elements(9) in TEST()
56 .pooling_elements(9) in TEST()
65 for (size_t pooling_elements = 2; pooling_elements < 9; pooling_elements++) { in TEST() local
67 .pooling_elements(pooling_elements) in TEST()
76 for (size_t pooling_elements = 2; pooling_elements < 9; pooling_elements++) { in TEST() local
78 .pooling_elements(pooling_elements) in TEST()
90 .pooling_elements(9) in TEST()
101 .pooling_elements(9) in TEST()
[all …]
|
| D | u8-maxpool-minmax.cc | 27 .pooling_elements(9) in TEST()
36 .pooling_elements(9) in TEST()
46 .pooling_elements(9) in TEST()
56 .pooling_elements(9) in TEST()
65 for (size_t pooling_elements = 2; pooling_elements < 9; pooling_elements++) { in TEST() local
67 .pooling_elements(pooling_elements) in TEST()
76 for (size_t pooling_elements = 2; pooling_elements < 9; pooling_elements++) { in TEST() local
78 .pooling_elements(pooling_elements) in TEST()
90 .pooling_elements(9) in TEST()
101 .pooling_elements(9) in TEST()
[all …]
|
| D | qu8-avgpool-minmax.cc | 28 .pooling_elements(17) in TEST()
37 .pooling_elements(17) in TEST()
48 .pooling_elements(17) in TEST()
61 .pooling_elements(17) in TEST()
73 .pooling_elements(17) in TEST()
85 .pooling_elements(17) in TEST()
97 .pooling_elements(17) in TEST()
108 .pooling_elements(17) in TEST()
118 .pooling_elements(17) in TEST()
127 for (size_t pooling_elements = 10; pooling_elements < 17; pooling_elements++) { in TEST() local
[all …]
|
| D | avgpool-microkernel-tester.h | 76 inline AvgPoolMicrokernelTester& pooling_elements(size_t pooling_elements) { in pooling_elements() function
77 assert(pooling_elements != 0); in pooling_elements()
78 this->pooling_elements_ = pooling_elements; in pooling_elements()
82 inline size_t pooling_elements() const { in pooling_elements() function
87 if (pooling_elements() <= primary_pooling_tile()) { in packed_pooling_elements()
90 …return (pooling_elements() - primary_pooling_tile()) % incremental_pooling_tile() == 0 ? pooling_e… in packed_pooling_elements()
234 for (size_t i = 0; i < (output_pixels() - 1) * step() + pooling_elements(); i++) {
238 indirect_input.begin() + (output_pixels() - 1) * step() + pooling_elements(), rng);
248 -int32_t(input_zero_point()) * int32_t(pooling_elements()),
249 input_scale() / (output_scale() * float(pooling_elements())),
[all …]
|
| D | argmaxpool-microkernel-tester.h | 61 inline ArgMaxPoolMicrokernelTester& pooling_elements(size_t pooling_elements) { in pooling_elements() function
62 assert(pooling_elements != 0); in pooling_elements()
63 this->pooling_elements_ = pooling_elements; in pooling_elements()
67 inline size_t pooling_elements() const { in pooling_elements() function
72 if (pooling_elements() <= primary_pooling_tile()) { in packed_pooling_elements()
75 …return (pooling_elements() - primary_pooling_tile()) % incremental_pooling_tile() == 0 ? pooling_e… in packed_pooling_elements()
154 ((output_pixels() - 1) * step() + pooling_elements()) * channels());
163 for (size_t i = 0; i < (output_pixels() - 1) * step() + pooling_elements(); i++) {
167 indirect_input.begin() + (output_pixels() - 1) * step() + pooling_elements(), rng);
174 for (size_t p = 0; p < pooling_elements(); p++) {
[all …]
|
| D | maxpool-microkernel-tester.h | 64 inline MaxPoolMicrokernelTester& pooling_elements(size_t pooling_elements) { in pooling_elements() argument
65 assert(pooling_elements != 0); in pooling_elements()
66 this->pooling_elements_ = pooling_elements; in pooling_elements()
70 inline size_t pooling_elements() const { in pooling_elements() function
75 if (pooling_elements() <= primary_pooling_tile()) { in packed_pooling_elements()
78 …return (pooling_elements() - primary_pooling_tile()) % incremental_pooling_tile() == 0 ? pooling_e… in packed_pooling_elements()
178 for (size_t i = 0; i < (output_pixels() - 1) * step() + pooling_elements(); i++) {
182 indirect_input.begin() + (output_pixels() - 1) * step() + pooling_elements(), rng);
199 for (size_t p = 0; p < pooling_elements(); p++) {
209 maxpool(output_pixels(), pooling_elements(), channels(),
[all …]
|
| /external/XNNPACK/src/f32-argmaxpool/ |
| D | 9x-scalar-c1.c | 14 size_t pooling_elements, in xnn_f32_argmaxpool_ukernel_9x__scalar_c1() argument
24 assert(pooling_elements != 0); in xnn_f32_argmaxpool_ukernel_9x__scalar_c1()
25 assert(pooling_elements <= 9); in xnn_f32_argmaxpool_ukernel_9x__scalar_c1()
47 if (pooling_elements < 2) { in xnn_f32_argmaxpool_ukernel_9x__scalar_c1()
50 if (pooling_elements <= 2) { in xnn_f32_argmaxpool_ukernel_9x__scalar_c1()
53 if (pooling_elements < 4) { in xnn_f32_argmaxpool_ukernel_9x__scalar_c1()
56 if (pooling_elements <= 4) { in xnn_f32_argmaxpool_ukernel_9x__scalar_c1()
59 if (pooling_elements < 6) { in xnn_f32_argmaxpool_ukernel_9x__scalar_c1()
62 if (pooling_elements <= 6) { in xnn_f32_argmaxpool_ukernel_9x__scalar_c1()
65 if (pooling_elements < 8) { in xnn_f32_argmaxpool_ukernel_9x__scalar_c1()
[all …]
|
| D | 4x-scalar-c1.c | 14 size_t pooling_elements, in xnn_f32_argmaxpool_ukernel_4x__scalar_c1() argument
24 assert(pooling_elements != 0); in xnn_f32_argmaxpool_ukernel_4x__scalar_c1()
25 assert(pooling_elements <= 4); in xnn_f32_argmaxpool_ukernel_4x__scalar_c1()
37 if (pooling_elements < 2) { in xnn_f32_argmaxpool_ukernel_4x__scalar_c1()
40 if (pooling_elements <= 2) { in xnn_f32_argmaxpool_ukernel_4x__scalar_c1()
43 if (pooling_elements != 4) { in xnn_f32_argmaxpool_ukernel_4x__scalar_c1()
|
| D | 9x-wasmsimd-c4.c | 15 size_t pooling_elements, in xnn_f32_argmaxpool_ukernel_9x__wasmsimd_c4() argument
25 assert(pooling_elements != 0); in xnn_f32_argmaxpool_ukernel_9x__wasmsimd_c4()
26 assert(pooling_elements <= 9); in xnn_f32_argmaxpool_ukernel_9x__wasmsimd_c4()
49 if (pooling_elements < 2) { in xnn_f32_argmaxpool_ukernel_9x__wasmsimd_c4()
52 if (pooling_elements <= 2) { in xnn_f32_argmaxpool_ukernel_9x__wasmsimd_c4()
55 if (pooling_elements < 4) { in xnn_f32_argmaxpool_ukernel_9x__wasmsimd_c4()
58 if (pooling_elements <= 4) { in xnn_f32_argmaxpool_ukernel_9x__wasmsimd_c4()
61 if (pooling_elements < 6) { in xnn_f32_argmaxpool_ukernel_9x__wasmsimd_c4()
64 if (pooling_elements <= 6) { in xnn_f32_argmaxpool_ukernel_9x__wasmsimd_c4()
67 if (pooling_elements < 8) { in xnn_f32_argmaxpool_ukernel_9x__wasmsimd_c4()
[all …]
|
| D | 9x-neon-c4.c | 15 size_t pooling_elements, in xnn_f32_argmaxpool_ukernel_9x__neon_c4() argument
25 assert(pooling_elements != 0); in xnn_f32_argmaxpool_ukernel_9x__neon_c4()
26 assert(pooling_elements <= 9); in xnn_f32_argmaxpool_ukernel_9x__neon_c4()
48 if (pooling_elements < 2) { in xnn_f32_argmaxpool_ukernel_9x__neon_c4()
51 if (pooling_elements <= 2) { in xnn_f32_argmaxpool_ukernel_9x__neon_c4()
54 if (pooling_elements < 4) { in xnn_f32_argmaxpool_ukernel_9x__neon_c4()
57 if (pooling_elements <= 4) { in xnn_f32_argmaxpool_ukernel_9x__neon_c4()
60 if (pooling_elements < 6) { in xnn_f32_argmaxpool_ukernel_9x__neon_c4()
63 if (pooling_elements <= 6) { in xnn_f32_argmaxpool_ukernel_9x__neon_c4()
66 if (pooling_elements < 8) { in xnn_f32_argmaxpool_ukernel_9x__neon_c4()
[all …]
|
| D | 9x-sse2-c4.c | 15 size_t pooling_elements, in xnn_f32_argmaxpool_ukernel_9x__sse2_c4() argument
25 assert(pooling_elements != 0); in xnn_f32_argmaxpool_ukernel_9x__sse2_c4()
26 assert(pooling_elements <= 9); in xnn_f32_argmaxpool_ukernel_9x__sse2_c4()
48 if (pooling_elements < 2) { in xnn_f32_argmaxpool_ukernel_9x__sse2_c4()
51 if (pooling_elements <= 2) { in xnn_f32_argmaxpool_ukernel_9x__sse2_c4()
54 if (pooling_elements < 4) { in xnn_f32_argmaxpool_ukernel_9x__sse2_c4()
57 if (pooling_elements <= 4) { in xnn_f32_argmaxpool_ukernel_9x__sse2_c4()
60 if (pooling_elements < 6) { in xnn_f32_argmaxpool_ukernel_9x__sse2_c4()
63 if (pooling_elements <= 6) { in xnn_f32_argmaxpool_ukernel_9x__sse2_c4()
66 if (pooling_elements < 8) { in xnn_f32_argmaxpool_ukernel_9x__sse2_c4()
[all …]
|
| D | 4x-wasmsimd-c4.c | 15 size_t pooling_elements, in xnn_f32_argmaxpool_ukernel_4x__wasmsimd_c4() argument
25 assert(pooling_elements != 0); in xnn_f32_argmaxpool_ukernel_4x__wasmsimd_c4()
26 assert(pooling_elements <= 4); in xnn_f32_argmaxpool_ukernel_4x__wasmsimd_c4()
39 if (pooling_elements < 2) { in xnn_f32_argmaxpool_ukernel_4x__wasmsimd_c4()
42 if (pooling_elements <= 2) { in xnn_f32_argmaxpool_ukernel_4x__wasmsimd_c4()
45 if (pooling_elements != 4) { in xnn_f32_argmaxpool_ukernel_4x__wasmsimd_c4()
|
| D | 4x-neon-c4.c | 15 size_t pooling_elements, in xnn_f32_argmaxpool_ukernel_4x__neon_c4() argument
25 assert(pooling_elements != 0); in xnn_f32_argmaxpool_ukernel_4x__neon_c4()
26 assert(pooling_elements <= 4); in xnn_f32_argmaxpool_ukernel_4x__neon_c4()
38 if (pooling_elements < 2) { in xnn_f32_argmaxpool_ukernel_4x__neon_c4()
41 if (pooling_elements <= 2) { in xnn_f32_argmaxpool_ukernel_4x__neon_c4()
44 if (pooling_elements != 4) { in xnn_f32_argmaxpool_ukernel_4x__neon_c4()
|
| D | 4x-sse2-c4.c | 15 size_t pooling_elements, in xnn_f32_argmaxpool_ukernel_4x__sse2_c4() argument
25 assert(pooling_elements != 0); in xnn_f32_argmaxpool_ukernel_4x__sse2_c4()
26 assert(pooling_elements <= 4); in xnn_f32_argmaxpool_ukernel_4x__sse2_c4()
38 if (pooling_elements < 2) { in xnn_f32_argmaxpool_ukernel_4x__sse2_c4()
41 if (pooling_elements <= 2) { in xnn_f32_argmaxpool_ukernel_4x__sse2_c4()
44 if (pooling_elements != 4) { in xnn_f32_argmaxpool_ukernel_4x__sse2_c4()
|
| D | 9p8x-scalar-c1.c | 14 size_t pooling_elements, in xnn_f32_argmaxpool_ukernel_9p8x__scalar_c1() argument
26 assert(pooling_elements != 0); in xnn_f32_argmaxpool_ukernel_9p8x__scalar_c1()
27 assert(pooling_elements > 9); in xnn_f32_argmaxpool_ukernel_9p8x__scalar_c1()
114 size_t k = pooling_elements; in xnn_f32_argmaxpool_ukernel_9p8x__scalar_c1()
|
| D | 9p8x-neon-c4.c | 15 size_t pooling_elements, in xnn_f32_argmaxpool_ukernel_9p8x__neon_c4() argument
27 assert(pooling_elements != 0); in xnn_f32_argmaxpool_ukernel_9p8x__neon_c4()
28 assert(pooling_elements > 9); in xnn_f32_argmaxpool_ukernel_9p8x__neon_c4()
109 size_t k = pooling_elements; in xnn_f32_argmaxpool_ukernel_9p8x__neon_c4()
|
| D | 9p8x-wasmsimd-c4.c | 15 size_t pooling_elements, in xnn_f32_argmaxpool_ukernel_9p8x__wasmsimd_c4() argument
27 assert(pooling_elements != 0); in xnn_f32_argmaxpool_ukernel_9p8x__wasmsimd_c4()
28 assert(pooling_elements > 9); in xnn_f32_argmaxpool_ukernel_9p8x__wasmsimd_c4()
120 size_t k = pooling_elements; in xnn_f32_argmaxpool_ukernel_9p8x__wasmsimd_c4()
|
| D | 9p8x-sse2-c4.c | 15 size_t pooling_elements, in xnn_f32_argmaxpool_ukernel_9p8x__sse2_c4() argument
27 assert(pooling_elements != 0); in xnn_f32_argmaxpool_ukernel_9p8x__sse2_c4()
28 assert(pooling_elements > 9); in xnn_f32_argmaxpool_ukernel_9p8x__sse2_c4()
120 size_t k = pooling_elements; in xnn_f32_argmaxpool_ukernel_9p8x__sse2_c4()
|