blob: 9a5c040aa3c923d0311b1c1967a29f367747620c [file] [log] [blame]
// Copyright 2019 Google LLC
//
// This source code is licensed under the BSD-style license found in the
// LICENSE file in the root directory of this source tree.
#include <assert.h>
#include <xnnpack/gavgpool.h>
#include <xnnpack/math.h>
void xnn_f32_gavgpool_minmax_ukernel_7p7x__wasm_c1(
size_t rows,
size_t channels,
const float* input,
size_t input_stride,
const float* zero,
float* buffer,
float* output,
const union xnn_f32_scaleminmax_params params[restrict XNN_MIN_ELEMENTS(1)])
{
assert(rows > 7);
assert(channels != 0);
const float* i0 = input;
const float* i1 = (const float*) ((uintptr_t) i0 + input_stride);
const float* i2 = (const float*) ((uintptr_t) i1 + input_stride);
const float* i3 = (const float*) ((uintptr_t) i2 + input_stride);
const float* i4 = (const float*) ((uintptr_t) i3 + input_stride);
const float* i5 = (const float*) ((uintptr_t) i4 + input_stride);
const float* i6 = (const float*) ((uintptr_t) i5 + input_stride);
const size_t input_increment = 7 * input_stride - channels * sizeof(float);
float* b = buffer;
size_t c = channels;
do {
const float vi0 = *i0++;
const float vi1 = *i1++;
const float vi2 = *i2++;
const float vi3 = *i3++;
const float vi4 = *i4++;
const float vi5 = *i5++;
const float vi6 = *i6++;
const float vsum01 = vi0 + vi1;
const float vsum23 = vi2 + vi3;
const float vsum45 = vi4 + vi5;
const float vsum016 = vsum01 + vi6;
const float vsum2345 = vsum23 + vsum45;
const float vsum = vsum016 + vsum2345;
*b++ = vsum;
} while (--c != 0);
for (rows -= 7; rows > 7; rows -= 7) {
b = buffer;
i0 = (const float*) ((uintptr_t) i0 + input_increment);
i1 = (const float*) ((uintptr_t) i1 + input_increment);
i2 = (const float*) ((uintptr_t) i2 + input_increment);
i3 = (const float*) ((uintptr_t) i3 + input_increment);
i4 = (const float*) ((uintptr_t) i4 + input_increment);
i5 = (const float*) ((uintptr_t) i5 + input_increment);
i6 = (const float*) ((uintptr_t) i6 + input_increment);
size_t c = channels;
do {
const float vi0 = *i0++;
const float vi1 = *i1++;
const float vi2 = *i2++;
const float vi3 = *i3++;
const float vi4 = *i4++;
const float vi5 = *i5++;
const float vi6 = *i6++;
const float vacc = *b;
const float vsum01 = vi0 + vi1;
const float vsum23 = vi2 + vi3;
const float vsum45 = vi4 + vi5;
const float vsum6a = vi6 + vacc;
const float vsum0123 = vsum01 + vsum23;
const float vsum456a = vsum45 + vsum6a;
const float vsum = vsum0123 + vsum456a;
*b++ = vsum;
} while (--c != 0);
}
i0 = (const float*) ((uintptr_t) i0 + input_increment);
i1 = (const float*) ((uintptr_t) i1 + input_increment);
if (rows < 2) {
i1 = zero;
}
i2 = (const float*) ((uintptr_t) i2 + input_increment);
if (rows <= 2) {
i2 = zero;
}
i3 = (const float*) ((uintptr_t) i3 + input_increment);
if (rows < 4) {
i3 = zero;
}
i4 = (const float*) ((uintptr_t) i4 + input_increment);
if (rows <= 4) {
i4 = zero;
}
i5 = (const float*) ((uintptr_t) i5 + input_increment);
if (rows < 6) {
i5 = zero;
}
i6 = (const float*) ((uintptr_t) i6 + input_increment);
if (rows <= 6) {
i6 = zero;
}
const float vscale = params->scalar.scale;
const float vmin = params->scalar.min;
const float vmax = params->scalar.max;
b = buffer;
do {
const float vi0 = *i0++;
const float vi1 = *i1++;
const float vi2 = *i2++;
const float vi3 = *i3++;
const float vi4 = *i4++;
const float vi5 = *i5++;
const float vi6 = *i6++;
const float vacc = *b++;
const float vsum01 = vi0 + vi1;
const float vsum23 = vi2 + vi3;
const float vsum45 = vi4 + vi5;
const float vsum6a = vi6 + vacc;
const float vsum0123 = vsum01 + vsum23;
const float vsum456a = vsum45 + vsum6a;
const float vsum = vsum0123 + vsum456a;
float vout = vsum * vscale;
vout = __builtin_wasm_max_f32(vout, vmin);
vout = __builtin_wasm_min_f32(vout, vmax);
*output++ = vout;
} while (--channels != 0);
}