Home
last modified time | relevance | path

Searched refs:pooling_elements (Results 1 – 25 of 26) sorted by relevance

12

/external/XNNPACK/test/
Df32-argmaxpool.cc24 .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 …]
Df32-maxpool-minmax.cc27 .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 …]
Df32-pavgpool-minmax.cc28 .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 …]
Df32-avgpool-minmax.cc28 .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 …]
Df16-maxpool-minmax.cc27 .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 …]
Df16-avgpool-minmax.cc28 .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 …]
Df16-pavgpool-minmax.cc28 .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 …]
Dqu8-avgpool-minmax.cc28 .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 …]
Du8-maxpool-minmax.cc27 .pooling_elements(9) in TEST()
38 .pooling_elements(9) in TEST()
50 .pooling_elements(9) in TEST()
61 .pooling_elements(9) in TEST()
71 for (size_t pooling_elements = 2; pooling_elements < 9; pooling_elements++) { in TEST() local
73 .pooling_elements(pooling_elements) in TEST()
84 for (size_t pooling_elements = 2; pooling_elements < 9; pooling_elements++) { in TEST() local
86 .pooling_elements(pooling_elements) in TEST()
100 .pooling_elements(9) in TEST()
113 .pooling_elements(9) in TEST()
[all …]
Ds8-maxpool-minmax.cc27 .pooling_elements(2) in TEST()
38 .pooling_elements(2) in TEST()
50 .pooling_elements(2) in TEST()
61 .pooling_elements(2) in TEST()
71 for (size_t pooling_elements = 2; pooling_elements < 2; pooling_elements++) { in TEST() local
73 .pooling_elements(pooling_elements) in TEST()
84 for (size_t pooling_elements = 2; pooling_elements < 2; pooling_elements++) { in TEST() local
86 .pooling_elements(pooling_elements) in TEST()
100 .pooling_elements(2) in TEST()
113 .pooling_elements(2) in TEST()
[all …]
Davgpool-microkernel-tester.h72 inline AvgPoolMicrokernelTester& pooling_elements(size_t pooling_elements) { in pooling_elements() argument
73 assert(pooling_elements != 0); in pooling_elements()
74 this->pooling_elements_ = pooling_elements; in pooling_elements()
78 inline size_t pooling_elements() const { in pooling_elements() function
83 if (pooling_elements() <= primary_pooling_tile()) { in packed_pooling_elements()
86 …return (pooling_elements() - primary_pooling_tile()) % incremental_pooling_tile() == 0 ? pooling_e… in packed_pooling_elements()
226 for (size_t i = 0; i < (output_pixels() - 1) * step() + pooling_elements(); i++) { in Test()
230 indirect_input.begin() + (output_pixels() - 1) * step() + pooling_elements(), rng); in Test()
239 for (size_t p = 0; p < pooling_elements(); p++) { in Test()
245 output_ref[x * channels() + c] = acc / float(pooling_elements()); in Test()
[all …]
Dargmaxpool-microkernel-tester.h60 inline ArgMaxPoolMicrokernelTester& pooling_elements(size_t pooling_elements) { in pooling_elements() function
61 assert(pooling_elements != 0); in pooling_elements()
62 this->pooling_elements_ = pooling_elements; in pooling_elements()
66 inline size_t pooling_elements() const { in pooling_elements() function
71 if (pooling_elements() <= primary_pooling_tile()) { in packed_pooling_elements()
74 …return (pooling_elements() - primary_pooling_tile()) % incremental_pooling_tile() == 0 ? pooling_e… in packed_pooling_elements()
153 ((output_pixels() - 1) * step() + pooling_elements()) * channels());
162 for (size_t i = 0; i < (output_pixels() - 1) * step() + pooling_elements(); i++) {
166 indirect_input.begin() + (output_pixels() - 1) * step() + pooling_elements(), rng);
173 for (size_t p = 0; p < pooling_elements(); p++) {
[all …]
Dmaxpool-microkernel-tester.h60 inline MaxPoolMicrokernelTester& pooling_elements(size_t pooling_elements) { in pooling_elements() function
61 assert(pooling_elements != 0); in pooling_elements()
62 this->pooling_elements_ = pooling_elements; in pooling_elements()
66 inline size_t pooling_elements() const { in pooling_elements() function
71 if (pooling_elements() <= primary_pooling_tile()) { in packed_pooling_elements()
74 …return (pooling_elements() - primary_pooling_tile()) % incremental_pooling_tile() == 0 ? pooling_e… in packed_pooling_elements()
179 for (size_t i = 0; i < (output_pixels() - 1) * step() + pooling_elements(); i++) { in Test()
183 indirect_input.begin() + (output_pixels() - 1) * step() + pooling_elements(), rng); in Test()
193 for (size_t p = 0; p < pooling_elements(); p++) { in Test()
203 maxpool(output_pixels(), pooling_elements(), channels(), in Test()
[all …]
/external/XNNPACK/src/f32-argmaxpool/
D9x-scalar-c1.c14 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 …]
D4x-scalar-c1.c14 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()
D9x-wasmsimd-c4.c15 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 …]
D9x-neon-c4.c15 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 …]
D9x-sse2-c4.c15 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 …]
D4x-wasmsimd-c4.c15 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()
D4x-neon-c4.c15 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()
D4x-sse2-c4.c15 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()
D9p8x-scalar-c1.c14 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()
D9p8x-neon-c4.c15 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()
D9p8x-wasmsimd-c4.c15 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()
D9p8x-sse2-c4.c15 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()

12