// 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 #include #include "average-pooling-operator-tester.h" TEST(AVERAGE_POOLING_NHWC_Q8, unit_batch_small_1xM_pool) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 2; pool_size <= xnn_params.q8.avgpool.mr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .TestQ8(); } } } TEST(AVERAGE_POOLING_NHWC_Q8, unit_batch_small_1xM_pool_with_padding) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 3; pool_size <= xnn_params.q8.avgpool.mr; pool_size++) { for (size_t padding_left = 0; padding_left <= 1; padding_left++) { for (size_t padding_right = 0; padding_right <= 1; padding_right++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .padding_left(padding_left) .padding_right(padding_right) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .TestQ8(); } } } } } TEST(AVERAGE_POOLING_NHWC_Q8, unit_batch_small_1xM_pool_with_stride) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 2; pool_size <= xnn_params.q8.avgpool.mr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 4) .pooling_height(1) .pooling_width(pool_size) .stride_width(2) .channels(channels) .TestQ8(); } } } TEST(AVERAGE_POOLING_NHWC_Q8, unit_batch_small_Mx1_pool) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 2; pool_size <= xnn_params.q8.avgpool.mr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .TestQ8(); } } } TEST(AVERAGE_POOLING_NHWC_Q8, unit_batch_small_Mx1_pool_with_padding) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 2; pool_size <= xnn_params.q8.avgpool.mr; pool_size++) { for (size_t padding_top = 0; padding_top <= 1; padding_top++) { for (size_t padding_bottom = 0; padding_bottom <= 1; padding_bottom++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .padding_top(padding_top) .padding_bottom(padding_bottom) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .TestQ8(); } } } } } TEST(AVERAGE_POOLING_NHWC_Q8, unit_batch_small_Mx1_pool_with_stride) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 2; pool_size <= xnn_params.q8.avgpool.mr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 3) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .stride_height(2) .channels(channels) .TestQ8(); } } } TEST(AVERAGE_POOLING_NHWC_Q8, unit_batch_small_pool_with_input_stride) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 2; pool_size <= xnn_params.q8.avgpool.mr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .input_pixel_stride(5 * channels) .TestQ8(); AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .input_pixel_stride(5 * channels) .TestQ8(); } } } TEST(AVERAGE_POOLING_NHWC_Q8, unit_batch_small_pool_with_output_stride) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 2; pool_size <= xnn_params.q8.avgpool.mr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .output_pixel_stride(5 * channels) .TestQ8(); AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .output_pixel_stride(5 * channels) .TestQ8(); } } } TEST(AVERAGE_POOLING_NHWC_Q8, unit_batch_small_pool_with_input_scale) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 2; pool_size <= xnn_params.q8.avgpool.mr; pool_size++) { for (float input_scale = 0.01f; input_scale < 100.0f; input_scale *= 3.14159265f) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .input_scale(input_scale) .TestQ8(); AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .input_scale(input_scale) .TestQ8(); } } } } TEST(AVERAGE_POOLING_NHWC_Q8, unit_batch_small_pool_with_input_zero_point) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 2; pool_size <= xnn_params.q8.avgpool.mr; pool_size++) { for (int32_t input_zero_point = 0; input_zero_point <= 255; input_zero_point += 51) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .input_zero_point(uint8_t(input_zero_point)) .TestQ8(); AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .input_zero_point(uint8_t(input_zero_point)) .TestQ8(); } } } } TEST(AVERAGE_POOLING_NHWC_Q8, unit_batch_small_pool_with_output_scale) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 2; pool_size <= xnn_params.q8.avgpool.mr; pool_size++) { for (float output_scale = 0.01f; output_scale < 100.0f; output_scale *= 3.14159265f) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .output_scale(output_scale) .TestQ8(); AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .output_scale(output_scale) .TestQ8(); } } } } TEST(AVERAGE_POOLING_NHWC_Q8, unit_batch_small_pool_with_output_zero_point) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 2; pool_size <= xnn_params.q8.avgpool.mr; pool_size++) { for (int32_t output_zero_point = 0; output_zero_point <= 255; output_zero_point += 51) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .output_zero_point(uint8_t(output_zero_point)) .TestQ8(); AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .output_zero_point(uint8_t(output_zero_point)) .TestQ8(); } } } } TEST(AVERAGE_POOLING_NHWC_Q8, unit_batch_small_pool_with_qmin) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 2; pool_size <= xnn_params.q8.avgpool.mr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .qmin(128) .TestQ8(); AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .qmin(128) .TestQ8(); } } } TEST(AVERAGE_POOLING_NHWC_Q8, unit_batch_small_pool_with_qmax) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 2; pool_size <= xnn_params.q8.avgpool.mr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .qmax(128) .TestQ8(); AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .qmax(128) .TestQ8(); } } } TEST(AVERAGE_POOLING_NHWC_Q8, unit_batch_large_1xM_pool) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = xnn_params.q8.avgpool.mr + 1; pool_size <= xnn_params.q8.avgpool.mr + xnn_params.q8.avgpool.qr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .TestQ8(); } } } TEST(AVERAGE_POOLING_NHWC_Q8, unit_batch_large_1xM_pool_with_padding) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 3; pool_size <= xnn_params.q8.avgpool.mr; pool_size++) { for (size_t padding_left = 0; padding_left <= 1; padding_left++) { for (size_t padding_right = 0; padding_right <= 1; padding_right++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .padding_left(padding_left) .padding_right(padding_right) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .TestQ8(); } } } } } TEST(AVERAGE_POOLING_NHWC_Q8, unit_batch_large_1xM_pool_with_stride) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = xnn_params.q8.avgpool.mr + 1; pool_size <= xnn_params.q8.avgpool.mr + xnn_params.q8.avgpool.qr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 4) .pooling_height(1) .pooling_width(pool_size) .stride_width(2) .channels(channels) .TestQ8(); } } } TEST(AVERAGE_POOLING_NHWC_Q8, unit_batch_large_Mx1_pool) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = xnn_params.q8.avgpool.mr + 1; pool_size <= xnn_params.q8.avgpool.mr + xnn_params.q8.avgpool.qr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .TestQ8(); } } } TEST(AVERAGE_POOLING_NHWC_Q8, unit_batch_large_Mx1_pool_with_padding) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = xnn_params.q8.avgpool.mr + 1; pool_size <= xnn_params.q8.avgpool.mr + xnn_params.q8.avgpool.qr; pool_size++) { for (size_t padding_top = 0; padding_top <= 1; padding_top++) { for (size_t padding_bottom = 0; padding_bottom <= 1; padding_bottom++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .padding_top(padding_top) .padding_bottom(padding_bottom) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .TestQ8(); } } } } } TEST(AVERAGE_POOLING_NHWC_Q8, unit_batch_large_Mx1_pool_with_stride) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = xnn_params.q8.avgpool.mr + 1; pool_size <= xnn_params.q8.avgpool.mr + xnn_params.q8.avgpool.qr; pool_size++) { for (size_t padding_top = 0; padding_top <= 1; padding_top++) { for (size_t padding_bottom = 0; padding_bottom <= 1; padding_bottom++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .padding_top(padding_top) .padding_bottom(padding_bottom) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .TestQ8(); } } } } } TEST(AVERAGE_POOLING_NHWC_Q8, unit_batch_large_pool_with_input_stride) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = xnn_params.q8.avgpool.mr + 1; pool_size <= xnn_params.q8.avgpool.mr + xnn_params.q8.avgpool.qr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .input_pixel_stride(5 * channels) .TestQ8(); AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .input_pixel_stride(5 * channels) .TestQ8(); } } } TEST(AVERAGE_POOLING_NHWC_Q8, unit_batch_large_pool_with_input_scale) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = xnn_params.q8.avgpool.mr + 1; pool_size <= xnn_params.q8.avgpool.mr + xnn_params.q8.avgpool.qr; pool_size++) { for (float input_scale = 0.01f; input_scale < 100.0f; input_scale *= 3.14159265f) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .input_scale(input_scale) .TestQ8(); AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .input_scale(input_scale) .TestQ8(); } } } } TEST(AVERAGE_POOLING_NHWC_Q8, unit_batch_large_pool_with_input_zero_point) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = xnn_params.q8.avgpool.mr + 1; pool_size <= xnn_params.q8.avgpool.mr + xnn_params.q8.avgpool.qr; pool_size++) { for (int32_t input_zero_point = 0; input_zero_point <= 255; input_zero_point += 51) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .input_zero_point(uint8_t(input_zero_point)) .TestQ8(); AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .input_zero_point(uint8_t(input_zero_point)) .TestQ8(); } } } } TEST(AVERAGE_POOLING_NHWC_Q8, unit_batch_large_pool_with_output_stride) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = xnn_params.q8.avgpool.mr + 1; pool_size <= xnn_params.q8.avgpool.mr + xnn_params.q8.avgpool.qr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .output_pixel_stride(5 * channels) .TestQ8(); AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .output_pixel_stride(5 * channels) .TestQ8(); } } } TEST(AVERAGE_POOLING_NHWC_Q8, unit_batch_large_pool_with_output_scale) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = xnn_params.q8.avgpool.mr + 1; pool_size <= xnn_params.q8.avgpool.mr + xnn_params.q8.avgpool.qr; pool_size++) { for (float output_scale = 0.01f; output_scale < 100.0f; output_scale *= 3.14159265f) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .output_scale(output_scale) .TestQ8(); AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .output_scale(output_scale) .TestQ8(); } } } } TEST(AVERAGE_POOLING_NHWC_Q8, unit_batch_large_pool_with_output_zero_point) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = xnn_params.q8.avgpool.mr + 1; pool_size <= xnn_params.q8.avgpool.mr + xnn_params.q8.avgpool.qr; pool_size++) { for (int32_t output_zero_point = 0; output_zero_point <= 255; output_zero_point += 51) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .output_zero_point(uint8_t(output_zero_point)) .TestQ8(); AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .output_zero_point(uint8_t(output_zero_point)) .TestQ8(); } } } } TEST(AVERAGE_POOLING_NHWC_Q8, unit_batch_large_pool_with_qmin) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = xnn_params.q8.avgpool.mr + 1; pool_size <= xnn_params.q8.avgpool.mr + xnn_params.q8.avgpool.qr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .qmin(128) .TestQ8(); AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .qmin(128) .TestQ8(); } } } TEST(AVERAGE_POOLING_NHWC_Q8, unit_batch_large_pool_with_qmax) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = xnn_params.q8.avgpool.mr + 1; pool_size <= xnn_params.q8.avgpool.mr + xnn_params.q8.avgpool.qr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .qmax(128) .TestQ8(); AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .qmax(128) .TestQ8(); } } } TEST(AVERAGE_POOLING_NHWC_Q8, small_batch_small_pool) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 2; pool_size <= xnn_params.q8.avgpool.mr; pool_size++) { AveragePoolingOperatorTester() .batch_size(3) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .TestQ8(); AveragePoolingOperatorTester() .batch_size(3) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .TestQ8(); } } } TEST(AVERAGE_POOLING_NHWC_Q8, small_batch_small_pool_with_input_stride) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 2; pool_size <= xnn_params.q8.avgpool.mr; pool_size++) { AveragePoolingOperatorTester() .batch_size(3) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .input_pixel_stride(5 * channels) .TestQ8(); AveragePoolingOperatorTester() .batch_size(3) .input_height(2) .input_width(pool_size + 1) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .input_pixel_stride(5 * channels) .TestQ8(); } } } TEST(AVERAGE_POOLING_NHWC_Q8, small_batch_small_pool_with_output_stride) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 2; pool_size <= xnn_params.q8.avgpool.mr; pool_size++) { AveragePoolingOperatorTester() .batch_size(3) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .output_pixel_stride(5 * channels) .TestQ8(); AveragePoolingOperatorTester() .batch_size(3) .input_height(2) .input_width(pool_size + 1) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .output_pixel_stride(5 * channels) .TestQ8(); } } } TEST(AVERAGE_POOLING_NHWC_Q8, small_batch_large_pool) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = xnn_params.q8.avgpool.mr + 1; pool_size <= xnn_params.q8.avgpool.mr + xnn_params.q8.avgpool.qr; pool_size++) { AveragePoolingOperatorTester() .batch_size(3) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .TestQ8(); AveragePoolingOperatorTester() .batch_size(3) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .TestQ8(); } } } TEST(AVERAGE_POOLING_NHWC_Q8, small_batch_large_pool_with_input_stride) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = xnn_params.q8.avgpool.mr + 1; pool_size <= xnn_params.q8.avgpool.mr + xnn_params.q8.avgpool.qr; pool_size++) { AveragePoolingOperatorTester() .batch_size(3) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .input_pixel_stride(5 * channels) .TestQ8(); AveragePoolingOperatorTester() .batch_size(3) .input_height(2) .input_width(pool_size + 1) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .input_pixel_stride(5 * channels) .TestQ8(); } } } TEST(AVERAGE_POOLING_NHWC_Q8, small_batch_large_pool_with_output_stride) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = xnn_params.q8.avgpool.mr + 1; pool_size <= xnn_params.q8.avgpool.mr + xnn_params.q8.avgpool.qr; pool_size++) { AveragePoolingOperatorTester() .batch_size(3) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .output_pixel_stride(5 * channels) .TestQ8(); AveragePoolingOperatorTester() .batch_size(3) .input_height(2) .input_width(pool_size + 1) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .output_pixel_stride(5 * channels) .TestQ8(); } } } TEST(AVERAGE_POOLING_NHWC_Q8, setup_increasing_batch) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); AveragePoolingOperatorTester() .batch_size(3) .next_batch_size(5) .input_height(8) .input_width(8) .pooling_height(5) .pooling_width(3) .channels(24) .TestSetupQ8(); } TEST(AVERAGE_POOLING_NHWC_Q8, setup_decreasing_batch) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); AveragePoolingOperatorTester() .batch_size(5) .next_batch_size(3) .input_height(8) .input_width(8) .pooling_height(5) .pooling_width(3) .channels(24) .TestSetupQ8(); } TEST(AVERAGE_POOLING_NHWC_Q8, setup_changing_height) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); AveragePoolingOperatorTester() .batch_size(3) .input_height(8) .input_width(8) .next_input_height(9) .pooling_height(5) .pooling_width(3) .channels(24) .TestSetupQ8(); AveragePoolingOperatorTester() .batch_size(3) .input_height(8) .input_width(8) .next_input_height(7) .pooling_height(5) .pooling_width(3) .channels(24) .TestSetupQ8(); } TEST(AVERAGE_POOLING_NHWC_Q8, setup_changing_width) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); AveragePoolingOperatorTester() .batch_size(3) .input_height(8) .input_width(8) .next_input_width(9) .pooling_height(5) .pooling_width(3) .channels(24) .TestSetupQ8(); AveragePoolingOperatorTester() .batch_size(3) .input_height(8) .input_width(8) .next_input_width(7) .pooling_height(5) .pooling_width(3) .channels(24) .TestSetupQ8(); } TEST(AVERAGE_POOLING_NHWC_Q8, setup_swap_height_and_width) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); AveragePoolingOperatorTester() .batch_size(3) .input_height(9) .input_width(8) .next_input_height(8) .next_input_width(9) .pooling_height(5) .pooling_width(3) .channels(24) .TestSetupQ8(); } TEST(AVERAGE_POOLING_NHWC_OP_F32, unit_batch_small_1xM_pool) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 2; pool_size <= xnn_params.f32.avgpool.mr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .TestF32(); } } } TEST(AVERAGE_POOLING_NHWC_OP_F32, unit_batch_small_1xM_pool_with_padding) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 3; pool_size <= xnn_params.f32.avgpool.mr; pool_size++) { for (size_t padding_left = 0; padding_left <= 1; padding_left++) { for (size_t padding_right = 0; padding_right <= 1; padding_right++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .padding_left(padding_left) .padding_right(padding_right) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .TestF32(); } } } } } TEST(AVERAGE_POOLING_NHWC_OP_F32, unit_batch_small_1xM_pool_with_stride) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 2; pool_size <= xnn_params.f32.avgpool.mr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 4) .pooling_height(1) .pooling_width(pool_size) .stride_width(2) .channels(channels) .TestF32(); } } } TEST(AVERAGE_POOLING_NHWC_OP_F32, unit_batch_small_Mx1_pool) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 2; pool_size <= xnn_params.f32.avgpool.mr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .TestF32(); } } } TEST(AVERAGE_POOLING_NHWC_OP_F32, unit_batch_small_Mx1_pool_with_padding) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 2; pool_size <= xnn_params.f32.avgpool.mr; pool_size++) { for (size_t padding_top = 0; padding_top <= 1; padding_top++) { for (size_t padding_bottom = 0; padding_bottom <= 1; padding_bottom++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .padding_top(padding_top) .padding_bottom(padding_bottom) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .TestF32(); } } } } } TEST(AVERAGE_POOLING_NHWC_OP_F32, unit_batch_small_Mx1_pool_with_stride) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 2; pool_size <= xnn_params.f32.avgpool.mr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 3) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .stride_height(2) .channels(channels) .TestF32(); } } } TEST(AVERAGE_POOLING_NHWC_OP_F32, unit_batch_small_pool_with_input_stride) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 2; pool_size <= xnn_params.f32.avgpool.mr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .input_pixel_stride(5 * channels) .TestF32(); AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .input_pixel_stride(5 * channels) .TestF32(); } } } TEST(AVERAGE_POOLING_NHWC_OP_F32, unit_batch_small_pool_with_output_stride) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 2; pool_size <= xnn_params.f32.avgpool.mr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .output_pixel_stride(5 * channels) .TestF32(); AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .output_pixel_stride(5 * channels) .TestF32(); } } } TEST(AVERAGE_POOLING_NHWC_OP_F32, unit_batch_small_pool_with_qmin) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 2; pool_size <= xnn_params.f32.avgpool.mr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .qmin(128) .TestF32(); AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .qmin(128) .TestF32(); } } } TEST(AVERAGE_POOLING_NHWC_OP_F32, unit_batch_small_pool_with_qmax) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 2; pool_size <= xnn_params.f32.avgpool.mr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .qmax(128) .TestF32(); AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .qmax(128) .TestF32(); } } } TEST(AVERAGE_POOLING_NHWC_OP_F32, unit_batch_large_1xM_pool) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = xnn_params.f32.avgpool.mr + 1; pool_size <= xnn_params.f32.avgpool.mr + xnn_params.f32.avgpool.qr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .TestF32(); } } } TEST(AVERAGE_POOLING_NHWC_OP_F32, unit_batch_large_1xM_pool_with_padding) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 3; pool_size <= xnn_params.f32.avgpool.mr; pool_size++) { for (size_t padding_left = 0; padding_left <= 1; padding_left++) { for (size_t padding_right = 0; padding_right <= 1; padding_right++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .padding_left(padding_left) .padding_right(padding_right) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .TestF32(); } } } } } TEST(AVERAGE_POOLING_NHWC_OP_F32, unit_batch_large_1xM_pool_with_stride) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = xnn_params.f32.avgpool.mr + 1; pool_size <= xnn_params.f32.avgpool.mr + xnn_params.f32.avgpool.qr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 4) .pooling_height(1) .pooling_width(pool_size) .stride_width(2) .channels(channels) .TestF32(); } } } TEST(AVERAGE_POOLING_NHWC_OP_F32, unit_batch_large_Mx1_pool) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = xnn_params.f32.avgpool.mr + 1; pool_size <= xnn_params.f32.avgpool.mr + xnn_params.f32.avgpool.qr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .TestF32(); } } } TEST(AVERAGE_POOLING_NHWC_OP_F32, unit_batch_large_Mx1_pool_with_padding) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = xnn_params.f32.avgpool.mr + 1; pool_size <= xnn_params.f32.avgpool.mr + xnn_params.f32.avgpool.qr; pool_size++) { for (size_t padding_top = 0; padding_top <= 1; padding_top++) { for (size_t padding_bottom = 0; padding_bottom <= 1; padding_bottom++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .padding_top(padding_top) .padding_bottom(padding_bottom) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .TestF32(); } } } } } TEST(AVERAGE_POOLING_NHWC_OP_F32, unit_batch_large_Mx1_pool_with_stride) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = xnn_params.f32.avgpool.mr + 1; pool_size <= xnn_params.f32.avgpool.mr + xnn_params.f32.avgpool.qr; pool_size++) { for (size_t padding_top = 0; padding_top <= 1; padding_top++) { for (size_t padding_bottom = 0; padding_bottom <= 1; padding_bottom++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .padding_top(padding_top) .padding_bottom(padding_bottom) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .TestF32(); } } } } } TEST(AVERAGE_POOLING_NHWC_OP_F32, unit_batch_large_pool_with_input_stride) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = xnn_params.f32.avgpool.mr + 1; pool_size <= xnn_params.f32.avgpool.mr + xnn_params.f32.avgpool.qr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .input_pixel_stride(5 * channels) .TestF32(); AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .input_pixel_stride(5 * channels) .TestF32(); } } } TEST(AVERAGE_POOLING_NHWC_OP_F32, unit_batch_large_pool_with_output_stride) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = xnn_params.f32.avgpool.mr + 1; pool_size <= xnn_params.f32.avgpool.mr + xnn_params.f32.avgpool.qr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .output_pixel_stride(5 * channels) .TestF32(); AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .output_pixel_stride(5 * channels) .TestF32(); } } } TEST(AVERAGE_POOLING_NHWC_OP_F32, unit_batch_large_pool_with_qmin) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = xnn_params.f32.avgpool.mr + 1; pool_size <= xnn_params.f32.avgpool.mr + xnn_params.f32.avgpool.qr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .qmin(128) .TestF32(); AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .qmin(128) .TestF32(); } } } TEST(AVERAGE_POOLING_NHWC_OP_F32, unit_batch_large_pool_with_qmax) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = xnn_params.f32.avgpool.mr + 1; pool_size <= xnn_params.f32.avgpool.mr + xnn_params.f32.avgpool.qr; pool_size++) { AveragePoolingOperatorTester() .batch_size(1) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .qmax(128) .TestF32(); AveragePoolingOperatorTester() .batch_size(1) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .qmax(128) .TestF32(); } } } TEST(AVERAGE_POOLING_NHWC_OP_F32, small_batch_small_pool) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 2; pool_size <= xnn_params.f32.avgpool.mr; pool_size++) { AveragePoolingOperatorTester() .batch_size(3) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .TestF32(); AveragePoolingOperatorTester() .batch_size(3) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .TestF32(); } } } TEST(AVERAGE_POOLING_NHWC_OP_F32, small_batch_small_pool_with_input_stride) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 2; pool_size <= xnn_params.f32.avgpool.mr; pool_size++) { AveragePoolingOperatorTester() .batch_size(3) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .input_pixel_stride(5 * channels) .TestF32(); AveragePoolingOperatorTester() .batch_size(3) .input_height(2) .input_width(pool_size + 1) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .input_pixel_stride(5 * channels) .TestF32(); } } } TEST(AVERAGE_POOLING_NHWC_OP_F32, small_batch_small_pool_with_output_stride) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = 2; pool_size <= xnn_params.f32.avgpool.mr; pool_size++) { AveragePoolingOperatorTester() .batch_size(3) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .output_pixel_stride(5 * channels) .TestF32(); AveragePoolingOperatorTester() .batch_size(3) .input_height(2) .input_width(pool_size + 1) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .output_pixel_stride(5 * channels) .TestF32(); } } } TEST(AVERAGE_POOLING_NHWC_OP_F32, small_batch_large_pool) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = xnn_params.f32.avgpool.mr + 1; pool_size <= xnn_params.f32.avgpool.mr + xnn_params.f32.avgpool.qr; pool_size++) { AveragePoolingOperatorTester() .batch_size(3) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .TestF32(); AveragePoolingOperatorTester() .batch_size(3) .input_height(2) .input_width(pool_size + 2) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .TestF32(); } } } TEST(AVERAGE_POOLING_NHWC_OP_F32, small_batch_large_pool_with_input_stride) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = xnn_params.f32.avgpool.mr + 1; pool_size <= xnn_params.f32.avgpool.mr + xnn_params.f32.avgpool.qr; pool_size++) { AveragePoolingOperatorTester() .batch_size(3) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .input_pixel_stride(5 * channels) .TestF32(); AveragePoolingOperatorTester() .batch_size(3) .input_height(2) .input_width(pool_size + 1) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .input_pixel_stride(5 * channels) .TestF32(); } } } TEST(AVERAGE_POOLING_NHWC_OP_F32, small_batch_large_pool_with_output_stride) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); for (size_t channels = 1; channels <= 100; channels += 15) { for (size_t pool_size = xnn_params.f32.avgpool.mr + 1; pool_size <= xnn_params.f32.avgpool.mr + xnn_params.f32.avgpool.qr; pool_size++) { AveragePoolingOperatorTester() .batch_size(3) .input_height(pool_size + 1) .input_width(3) .pooling_height(pool_size) .pooling_width(1) .channels(channels) .output_pixel_stride(5 * channels) .TestF32(); AveragePoolingOperatorTester() .batch_size(3) .input_height(2) .input_width(pool_size + 1) .pooling_height(1) .pooling_width(pool_size) .channels(channels) .output_pixel_stride(5 * channels) .TestF32(); } } } TEST(AVERAGE_POOLING_NHWC_OP_F32, setup_increasing_batch) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); AveragePoolingOperatorTester() .batch_size(3) .next_batch_size(5) .input_height(8) .input_width(8) .pooling_height(5) .pooling_width(3) .channels(24) .TestSetupF32(); } TEST(AVERAGE_POOLING_NHWC_OP_F32, setup_decreasing_batch) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); AveragePoolingOperatorTester() .batch_size(5) .next_batch_size(3) .input_height(8) .input_width(8) .pooling_height(5) .pooling_width(3) .channels(24) .TestSetupF32(); } TEST(AVERAGE_POOLING_NHWC_OP_F32, setup_changing_height) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); AveragePoolingOperatorTester() .batch_size(3) .input_height(8) .input_width(8) .next_input_height(9) .pooling_height(5) .pooling_width(3) .channels(24) .TestSetupF32(); AveragePoolingOperatorTester() .batch_size(3) .input_height(8) .input_width(8) .next_input_height(7) .pooling_height(5) .pooling_width(3) .channels(24) .TestSetupF32(); } TEST(AVERAGE_POOLING_NHWC_OP_F32, setup_changing_width) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); AveragePoolingOperatorTester() .batch_size(3) .input_height(8) .input_width(8) .next_input_width(9) .pooling_height(5) .pooling_width(3) .channels(24) .TestSetupF32(); AveragePoolingOperatorTester() .batch_size(3) .input_height(8) .input_width(8) .next_input_width(7) .pooling_height(5) .pooling_width(3) .channels(24) .TestSetupF32(); } TEST(AVERAGE_POOLING_NHWC_OP_F32, setup_swap_height_and_width) { ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */)); AveragePoolingOperatorTester() .batch_size(3) .input_height(9) .input_width(8) .next_input_height(8) .next_input_width(9) .pooling_height(5) .pooling_width(3) .channels(24) .TestSetupF32(); }