Initial open-source release
PiperOrigin-RevId: 271685289
diff --git a/test/global-average-pooling.cc b/test/global-average-pooling.cc
new file mode 100644
index 0000000..63cb487
--- /dev/null
+++ b/test/global-average-pooling.cc
@@ -0,0 +1,526 @@
+// Copyright (c) Facebook, Inc. and its affiliates.
+// All rights reserved.
+//
+// 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 <gtest/gtest.h>
+
+#include "global-average-pooling-operator-tester.h"
+
+#include <xnnpack/params.h>
+
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_Q8, unit_batch_small_width) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = 1; width <= xnn_params.q8.gavgpool.mr; width++) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(1)
+ .width(width)
+ .channels(channels)
+ .TestQ8();
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_Q8, unit_batch_small_width_with_input_stride) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = 1; width <= xnn_params.q8.gavgpool.mr; width++) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(1)
+ .width(width)
+ .channels(channels)
+ .input_stride(5 * channels)
+ .TestQ8();
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_Q8, unit_batch_small_width_with_input_scale) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = 1; width <= xnn_params.q8.gavgpool.mr; width++) {
+ for (float input_scale = 0.01f; input_scale < 100.0f; input_scale *= 3.14159265f) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(1)
+ .width(width)
+ .channels(channels)
+ .input_scale(input_scale)
+ .TestQ8();
+ }
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_Q8, unit_batch_small_width_with_input_zero_point) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = 1; width <= xnn_params.q8.gavgpool.mr; width++) {
+ for (int32_t input_zero_point = 0; input_zero_point <= 255; input_zero_point += 51) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(1)
+ .width(width)
+ .channels(channels)
+ .input_zero_point(uint8_t(input_zero_point))
+ .TestQ8();
+ }
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_Q8, unit_batch_small_width_with_output_scale) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = 1; width <= xnn_params.q8.gavgpool.mr; width++) {
+ for (float output_scale = 0.01f; output_scale < 100.0f; output_scale *= 3.14159265f) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(1)
+ .width(width)
+ .channels(channels)
+ .output_scale(output_scale)
+ .TestQ8();
+ }
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_Q8, unit_batch_small_width_with_output_zero_point) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = 1; width <= xnn_params.q8.gavgpool.mr; width++) {
+ for (int32_t output_zero_point = 0; output_zero_point <= 255; output_zero_point += 51) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(1)
+ .width(width)
+ .channels(channels)
+ .output_zero_point(uint8_t(output_zero_point))
+ .TestQ8();
+ }
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_Q8, unit_batch_small_width_with_qmin) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = 1; width <= xnn_params.q8.gavgpool.mr; width++) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(1)
+ .width(width)
+ .channels(channels)
+ .qmin(128)
+ .TestQ8();
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_Q8, unit_batch_small_width_with_qmax) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = 1; width <= xnn_params.q8.gavgpool.mr; width++) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(1)
+ .width(width)
+ .channels(channels)
+ .qmax(128)
+ .TestQ8();
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_Q8, unit_batch_large_width) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = xnn_params.q8.gavgpool.mr; width <= 4 * xnn_params.q8.gavgpool.mr; width++) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(1)
+ .width(width)
+ .channels(channels)
+ .TestQ8();
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_Q8, unit_batch_large_width_with_input_stride) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = xnn_params.q8.gavgpool.mr; width <= 4 * xnn_params.q8.gavgpool.mr; width++) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(1)
+ .width(width)
+ .channels(channels)
+ .input_stride(5 * channels)
+ .TestQ8();
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_Q8, unit_batch_large_width_with_input_scale) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = xnn_params.q8.gavgpool.mr; width <= 4 * xnn_params.q8.gavgpool.mr; width++) {
+ for (float input_scale = 0.01f; input_scale < 100.0f; input_scale *= 3.14159265f) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(1)
+ .width(width)
+ .channels(channels)
+ .input_scale(input_scale)
+ .TestQ8();
+ }
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_Q8, unit_batch_large_width_with_input_zero_point) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = xnn_params.q8.gavgpool.mr; width <= 4 * xnn_params.q8.gavgpool.mr; width++) {
+ for (int32_t input_zero_point = 0; input_zero_point <= 255; input_zero_point += 51) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(1)
+ .width(width)
+ .channels(channels)
+ .input_zero_point(uint8_t(input_zero_point))
+ .TestQ8();
+ }
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_Q8, unit_batch_large_width_with_output_scale) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = xnn_params.q8.gavgpool.mr; width <= 4 * xnn_params.q8.gavgpool.mr; width++) {
+ for (float output_scale = 0.01f; output_scale < 100.0f; output_scale *= 3.14159265f) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(1)
+ .width(width)
+ .channels(channels)
+ .output_scale(output_scale)
+ .TestQ8();
+ }
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_Q8, unit_batch_large_width_with_output_zero_point) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = xnn_params.q8.gavgpool.mr; width <= 4 * xnn_params.q8.gavgpool.mr; width++) {
+ for (int32_t output_zero_point = 0; output_zero_point <= 255; output_zero_point += 51) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(1)
+ .width(width)
+ .channels(channels)
+ .output_zero_point(uint8_t(output_zero_point))
+ .TestQ8();
+ }
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_Q8, unit_batch_large_width_with_qmin) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = xnn_params.q8.gavgpool.mr; width <= 4 * xnn_params.q8.gavgpool.mr; width++) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(1)
+ .width(width)
+ .channels(channels)
+ .qmin(128)
+ .TestQ8();
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_Q8, unit_batch_large_width_with_qmax) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = xnn_params.q8.gavgpool.mr; width <= 4 * xnn_params.q8.gavgpool.mr; width++) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(1)
+ .width(width)
+ .channels(channels)
+ .qmax(128)
+ .TestQ8();
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_Q8, small_batch_small_width) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = 1; width <= xnn_params.q8.gavgpool.mr; width++) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(3)
+ .width(width)
+ .channels(channels)
+ .TestQ8();
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_Q8, small_batch_small_width_with_input_stride) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = 1; width <= xnn_params.q8.gavgpool.mr; width++) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(3)
+ .width(width)
+ .channels(channels)
+ .input_stride(5 * channels)
+ .TestQ8();
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_Q8, small_batch_small_width_with_output_stride) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = 1; width <= xnn_params.q8.gavgpool.mr; width++) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(3)
+ .width(width)
+ .channels(channels)
+ .output_stride(5 * channels)
+ .TestQ8();
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_Q8, small_batch_large_width) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = xnn_params.q8.gavgpool.mr; width <= 4 * xnn_params.q8.gavgpool.mr; width++) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(3)
+ .width(width)
+ .channels(channels)
+ .TestQ8();
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_Q8, small_batch_large_width_with_input_stride) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = xnn_params.q8.gavgpool.mr; width <= 4 * xnn_params.q8.gavgpool.mr; width++) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(3)
+ .width(width)
+ .channels(channels)
+ .input_stride(5 * channels)
+ .TestQ8();
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_Q8, small_batch_large_width_with_output_stride) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = xnn_params.q8.gavgpool.mr; width <= 4 * xnn_params.q8.gavgpool.mr; width++) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(3)
+ .width(width)
+ .channels(channels)
+ .output_stride(5 * channels)
+ .TestQ8();
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_F32, unit_batch_small_width) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = 1; width <= xnn_params.f32.gavgpool.mr; width++) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(1)
+ .width(width)
+ .channels(channels)
+ .TestF32();
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_F32, unit_batch_small_width_with_input_stride) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = 1; width <= xnn_params.f32.gavgpool.mr; width++) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(1)
+ .width(width)
+ .channels(channels)
+ .input_stride(5 * channels)
+ .TestF32();
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_F32, unit_batch_small_width_with_qmin) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = 1; width <= xnn_params.f32.gavgpool.mr; width++) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(1)
+ .width(width)
+ .channels(channels)
+ .qmin(128)
+ .TestF32();
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_F32, unit_batch_small_width_with_qmax) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = 1; width <= xnn_params.f32.gavgpool.mr; width++) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(1)
+ .width(width)
+ .channels(channels)
+ .qmax(128)
+ .TestF32();
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_F32, unit_batch_large_width) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = xnn_params.f32.gavgpool.mr; width <= 4 * xnn_params.f32.gavgpool.mr; width++) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(1)
+ .width(width)
+ .channels(channels)
+ .TestF32();
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_F32, unit_batch_large_width_with_input_stride) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = xnn_params.f32.gavgpool.mr; width <= 4 * xnn_params.f32.gavgpool.mr; width++) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(1)
+ .width(width)
+ .channels(channels)
+ .input_stride(5 * channels)
+ .TestF32();
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_F32, unit_batch_large_width_with_qmin) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = xnn_params.f32.gavgpool.mr; width <= 4 * xnn_params.f32.gavgpool.mr; width++) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(1)
+ .width(width)
+ .channels(channels)
+ .qmin(128)
+ .TestF32();
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_F32, unit_batch_large_width_with_qmax) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = xnn_params.f32.gavgpool.mr; width <= 4 * xnn_params.f32.gavgpool.mr; width++) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(1)
+ .width(width)
+ .channels(channels)
+ .qmax(128)
+ .TestF32();
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_F32, small_batch_small_width) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = 1; width <= xnn_params.f32.gavgpool.mr; width++) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(3)
+ .width(width)
+ .channels(channels)
+ .TestF32();
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_F32, small_batch_small_width_with_input_stride) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = 1; width <= xnn_params.f32.gavgpool.mr; width++) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(3)
+ .width(width)
+ .channels(channels)
+ .input_stride(5 * channels)
+ .TestF32();
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_F32, small_batch_small_width_with_output_stride) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = 1; width <= xnn_params.f32.gavgpool.mr; width++) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(3)
+ .width(width)
+ .channels(channels)
+ .output_stride(5 * channels)
+ .TestF32();
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_F32, small_batch_large_width) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = xnn_params.f32.gavgpool.mr; width <= 4 * xnn_params.f32.gavgpool.mr; width++) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(3)
+ .width(width)
+ .channels(channels)
+ .TestF32();
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_F32, small_batch_large_width_with_input_stride) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = xnn_params.f32.gavgpool.mr; width <= 4 * xnn_params.f32.gavgpool.mr; width++) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(3)
+ .width(width)
+ .channels(channels)
+ .input_stride(5 * channels)
+ .TestF32();
+ }
+ }
+}
+
+TEST(GLOBAL_AVERAGE_POOLING_OP_F32, small_batch_large_width_with_output_stride) {
+ ASSERT_EQ(xnn_status_success, xnn_initialize());
+ for (size_t channels = 1; channels <= 100; channels += 15) {
+ for (size_t width = xnn_params.f32.gavgpool.mr; width <= 4 * xnn_params.f32.gavgpool.mr; width++) {
+ GlobalAveragePoolingOperatorTester()
+ .batch_size(3)
+ .width(width)
+ .channels(channels)
+ .output_stride(5 * channels)
+ .TestF32();
+ }
+ }
+}