1 // Copyright 2019 Google LLC
2 //
3 // This source code is licensed under the BSD-style license found in the
4 // LICENSE file in the root directory of this source tree.
5
6 #include <assert.h>
7
8 #include <xnnpack/argmaxpool.h>
9 #include <xnnpack/math.h>
10
11
xnn_f32_argmaxpool_ukernel_9x__scalar_c1(size_t output_pixels,size_t pooling_elements,size_t channels,const float ** input,size_t input_offset,float * output,uint32_t * index,size_t input_increment,size_t output_increment)12 void xnn_f32_argmaxpool_ukernel_9x__scalar_c1(
13 size_t output_pixels,
14 size_t pooling_elements,
15 size_t channels,
16 const float** input,
17 size_t input_offset,
18 float* output,
19 uint32_t* index,
20 size_t input_increment,
21 size_t output_increment)
22 {
23 assert(output_pixels != 0);
24 assert(pooling_elements != 0);
25 assert(pooling_elements <= 9);
26 assert(channels != 0);
27
28 do {
29 const float* i0 = input[0];
30 const float* i1 = input[1];
31 const float* i2 = input[2];
32 const float* i3 = input[3];
33 const float* i4 = input[4];
34 const float* i5 = input[5];
35 const float* i6 = input[6];
36 const float* i7 = input[7];
37 const float* i8 = input[8];
38 i0 = (const float*) ((uintptr_t) i0 + input_offset);
39 i1 = (const float*) ((uintptr_t) i1 + input_offset);
40 i2 = (const float*) ((uintptr_t) i2 + input_offset);
41 i3 = (const float*) ((uintptr_t) i3 + input_offset);
42 i4 = (const float*) ((uintptr_t) i4 + input_offset);
43 i5 = (const float*) ((uintptr_t) i5 + input_offset);
44 i6 = (const float*) ((uintptr_t) i6 + input_offset);
45 i7 = (const float*) ((uintptr_t) i7 + input_offset);
46 i8 = (const float*) ((uintptr_t) i8 + input_offset);
47 if (pooling_elements < 2) {
48 i1 = i0;
49 }
50 if (pooling_elements <= 2) {
51 i2 = i0;
52 }
53 if (pooling_elements < 4) {
54 i3 = i0;
55 }
56 if (pooling_elements <= 4) {
57 i4 = i0;
58 }
59 if (pooling_elements < 6) {
60 i5 = i0;
61 }
62 if (pooling_elements <= 6) {
63 i6 = i0;
64 }
65 if (pooling_elements < 8) {
66 i7 = i0;
67 }
68 if (pooling_elements <= 8) {
69 i8 = i0;
70 }
71
72 size_t c = channels;
73 do {
74 const float vi0 = *i0++;
75 const float vi1 = *i1++;
76 const float vi2 = *i2++;
77 const float vi3 = *i3++;
78 const float vi4 = *i4++;
79 const float vi5 = *i5++;
80 const float vi6 = *i6++;
81 const float vi7 = *i7++;
82 const float vi8 = *i8++;
83
84 float vmax = vi0;
85 uint32_t vidx = 0;
86
87 if (vi1 > vmax) {
88 vmax = vi1;
89 vidx = 1;
90 }
91
92 if (vi2 > vmax) {
93 vmax = vi2;
94 vidx = 2;
95 }
96
97 if (vi3 > vmax) {
98 vmax = vi3;
99 vidx = 3;
100 }
101
102 if (vi4 > vmax) {
103 vmax = vi4;
104 vidx = 4;
105 }
106
107 if (vi5 > vmax) {
108 vmax = vi5;
109 vidx = 5;
110 }
111
112 if (vi6 > vmax) {
113 vmax = vi6;
114 vidx = 6;
115 }
116
117 if (vi7 > vmax) {
118 vmax = vi7;
119 vidx = 7;
120 }
121
122 if (vi8 > vmax) {
123 vmax = vi8;
124 vidx = 8;
125 }
126
127 *output++ = vmax;
128 *index++ = vidx;
129 } while (--c != 0);
130 input = (const float**) ((uintptr_t) input + input_increment);
131 output = (float*) ((uintptr_t) output + output_increment);
132 } while (--output_pixels != 0);
133 }
134