3557 lines
118 KiB
C++
3557 lines
118 KiB
C++
// 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 <cassert>
|
|
#include <cstddef>
|
|
|
|
#include <gtest/gtest.h>
|
|
#include "xnnpack.h"
|
|
#include "xnnpack/config.h"
|
|
#include "max-pooling-operator-tester.h"
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, unit_batch_small_1xM_pool) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, unit_batch_small_1xM_pool_with_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 3; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
for (size_t padding_left = 0; padding_left <= 1; padding_left++) {
|
|
for (size_t padding_right = 0; padding_right <= 1; padding_right++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, unit_batch_small_1xM_pool_with_tf_same_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 3; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
for (size_t input_width = pool_size; input_width <= pool_size * 2; input_width++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(input_width)
|
|
.padding_tf_same(true)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, unit_batch_small_1xM_pool_with_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 4)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.stride_width(2)
|
|
.channels(channels)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, unit_batch_small_1xM_pool_with_dilation) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(2 * pool_size + 1)
|
|
.padding_left(1)
|
|
.padding_right(1)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.dilation_width(2)
|
|
.channels(channels)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, unit_batch_small_Mx1_pool) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, unit_batch_small_Mx1_pool_with_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
for (size_t padding_top = 0; padding_top <= 1; padding_top++) {
|
|
for (size_t padding_bottom = 0; padding_bottom <= 1; padding_bottom++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, unit_batch_small_Mx1_pool_with_tf_same_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
for (size_t input_height = pool_size; input_height <= pool_size * 2; input_height++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(input_height)
|
|
.input_width(3)
|
|
.padding_tf_same(true)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, unit_batch_small_Mx1_pool_with_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 3)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.stride_height(2)
|
|
.channels(channels)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, unit_batch_small_Mx1_pool_with_dilation) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2 * pool_size)
|
|
.input_width(3)
|
|
.padding_top(1)
|
|
.padding_bottom(1)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.dilation_height(2)
|
|
.channels(channels)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, unit_batch_small_pool_with_input_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestS8();
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, unit_batch_small_pool_with_output_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestS8();
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, unit_batch_small_pool_with_qmin) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.qmin(192)
|
|
.TestS8();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.qmin(192)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, unit_batch_small_pool_with_qmax) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.qmax(192)
|
|
.TestS8();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.qmax(192)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, unit_batch_large_1xM_pool) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, unit_batch_large_1xM_pool_with_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
for (size_t padding_left = 0; padding_left <= 1; padding_left++) {
|
|
for (size_t padding_right = 0; padding_right <= 1; padding_right++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, unit_batch_large_1xM_pool_with_tf_same_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
for (size_t input_width = pool_size; input_width <= pool_size * 2; input_width++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(input_width)
|
|
.padding_tf_same(true)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, unit_batch_large_1xM_pool_with_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 4)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.stride_width(2)
|
|
.channels(channels)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, unit_batch_large_1xM_pool_with_dilation) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(2 * pool_size + 1)
|
|
.padding_left(1)
|
|
.padding_right(1)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.dilation_width(2)
|
|
.channels(channels)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, unit_batch_large_Mx1_pool) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, unit_batch_large_Mx1_pool_with_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
for (size_t padding_top = 0; padding_top <= 1; padding_top++) {
|
|
for (size_t padding_bottom = 0; padding_bottom <= 1; padding_bottom++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, unit_batch_large_Mx1_pool_with_tf_same_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
for (size_t input_height = pool_size; input_height <= pool_size * 2; input_height++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(input_height)
|
|
.input_width(3)
|
|
.padding_tf_same(true)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, unit_batch_large_Mx1_pool_with_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 3)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.stride_height(2)
|
|
.channels(channels)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, unit_batch_large_Mx1_pool_with_dilation) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2 * pool_size)
|
|
.input_width(3)
|
|
.padding_top(1)
|
|
.padding_bottom(1)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.dilation_height(2)
|
|
.channels(channels)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, unit_batch_large_pool_with_input_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestS8();
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, unit_batch_large_pool_with_output_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestS8();
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, unit_batch_large_pool_with_qmin) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.qmin(192)
|
|
.TestS8();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.qmin(192)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, unit_batch_large_pool_with_qmax) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.qmax(192)
|
|
.TestS8();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.qmax(192)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, small_batch_small_pool) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.TestS8();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, small_batch_small_pool_with_input_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestS8();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.input_pixel_stride(5 * channels)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, small_batch_small_pool_with_output_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestS8();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.output_pixel_stride(5 * channels)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, small_batch_large_pool) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.TestS8();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, small_batch_large_pool_with_input_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestS8();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.input_pixel_stride(5 * channels)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, small_batch_large_pool_with_output_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_s8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestS8();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.output_pixel_stride(5 * channels)
|
|
.TestS8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, setup_increasing_batch) {
|
|
ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.next_batch_size(5)
|
|
.input_height(8)
|
|
.input_width(8)
|
|
.pooling_height(5)
|
|
.pooling_width(3)
|
|
.channels(24)
|
|
.TestSetupS8();
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, setup_decreasing_batch) {
|
|
ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(5)
|
|
.next_batch_size(3)
|
|
.input_height(8)
|
|
.input_width(8)
|
|
.pooling_height(5)
|
|
.pooling_width(3)
|
|
.channels(24)
|
|
.TestSetupS8();
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, setup_changing_height) {
|
|
ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(8)
|
|
.input_width(8)
|
|
.next_input_height(9)
|
|
.pooling_height(5)
|
|
.pooling_width(3)
|
|
.channels(24)
|
|
.TestSetupS8();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(8)
|
|
.input_width(8)
|
|
.next_input_height(7)
|
|
.pooling_height(5)
|
|
.pooling_width(3)
|
|
.channels(24)
|
|
.TestSetupS8();
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, setup_changing_width) {
|
|
ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(8)
|
|
.input_width(8)
|
|
.next_input_width(9)
|
|
.pooling_height(5)
|
|
.pooling_width(3)
|
|
.channels(24)
|
|
.TestSetupS8();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(8)
|
|
.input_width(8)
|
|
.next_input_width(7)
|
|
.pooling_height(5)
|
|
.pooling_width(3)
|
|
.channels(24)
|
|
.TestSetupS8();
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_S8, setup_swap_height_and_width) {
|
|
ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestSetupS8();
|
|
}
|
|
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, unit_batch_small_1xM_pool) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, unit_batch_small_1xM_pool_with_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 3; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
for (size_t padding_left = 0; padding_left <= 1; padding_left++) {
|
|
for (size_t padding_right = 0; padding_right <= 1; padding_right++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, unit_batch_small_1xM_pool_with_tf_same_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 3; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
for (size_t input_width = pool_size; input_width <= pool_size * 2; input_width++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(input_width)
|
|
.padding_tf_same(true)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, unit_batch_small_1xM_pool_with_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 4)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.stride_width(2)
|
|
.channels(channels)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, unit_batch_small_1xM_pool_with_dilation) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(2 * pool_size + 1)
|
|
.padding_left(1)
|
|
.padding_right(1)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.dilation_width(2)
|
|
.channels(channels)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, unit_batch_small_Mx1_pool) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, unit_batch_small_Mx1_pool_with_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
for (size_t padding_top = 0; padding_top <= 1; padding_top++) {
|
|
for (size_t padding_bottom = 0; padding_bottom <= 1; padding_bottom++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, unit_batch_small_Mx1_pool_with_tf_same_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
for (size_t input_height = pool_size; input_height <= pool_size * 2; input_height++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(input_height)
|
|
.input_width(3)
|
|
.padding_tf_same(true)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, unit_batch_small_Mx1_pool_with_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 3)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.stride_height(2)
|
|
.channels(channels)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, unit_batch_small_Mx1_pool_with_dilation) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2 * pool_size)
|
|
.input_width(3)
|
|
.padding_top(1)
|
|
.padding_bottom(1)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.dilation_height(2)
|
|
.channels(channels)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, unit_batch_small_pool_with_input_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestU8();
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, unit_batch_small_pool_with_output_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestU8();
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, unit_batch_small_pool_with_qmin) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.qmin(192)
|
|
.TestU8();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.qmin(192)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, unit_batch_small_pool_with_qmax) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.qmax(192)
|
|
.TestU8();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.qmax(192)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, unit_batch_large_1xM_pool) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, unit_batch_large_1xM_pool_with_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
for (size_t padding_left = 0; padding_left <= 1; padding_left++) {
|
|
for (size_t padding_right = 0; padding_right <= 1; padding_right++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, unit_batch_large_1xM_pool_with_tf_same_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
for (size_t input_width = pool_size; input_width <= pool_size * 2; input_width++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(input_width)
|
|
.padding_tf_same(true)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, unit_batch_large_1xM_pool_with_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 4)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.stride_width(2)
|
|
.channels(channels)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, unit_batch_large_1xM_pool_with_dilation) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(2 * pool_size + 1)
|
|
.padding_left(1)
|
|
.padding_right(1)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.dilation_width(2)
|
|
.channels(channels)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, unit_batch_large_Mx1_pool) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, unit_batch_large_Mx1_pool_with_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
for (size_t padding_top = 0; padding_top <= 1; padding_top++) {
|
|
for (size_t padding_bottom = 0; padding_bottom <= 1; padding_bottom++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, unit_batch_large_Mx1_pool_with_tf_same_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
for (size_t input_height = pool_size; input_height <= pool_size * 2; input_height++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(input_height)
|
|
.input_width(3)
|
|
.padding_tf_same(true)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, unit_batch_large_Mx1_pool_with_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 3)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.stride_height(2)
|
|
.channels(channels)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, unit_batch_large_Mx1_pool_with_dilation) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2 * pool_size)
|
|
.input_width(3)
|
|
.padding_top(1)
|
|
.padding_bottom(1)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.dilation_height(2)
|
|
.channels(channels)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, unit_batch_large_pool_with_input_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestU8();
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, unit_batch_large_pool_with_output_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestU8();
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, unit_batch_large_pool_with_qmin) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.qmin(192)
|
|
.TestU8();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.qmin(192)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, unit_batch_large_pool_with_qmax) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.qmax(192)
|
|
.TestU8();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.qmax(192)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, small_batch_small_pool) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.TestU8();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, small_batch_small_pool_with_input_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestU8();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.input_pixel_stride(5 * channels)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, small_batch_small_pool_with_output_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestU8();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.output_pixel_stride(5 * channels)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, small_batch_large_pool) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.TestU8();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, small_batch_large_pool_with_input_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestU8();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.input_pixel_stride(5 * channels)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, small_batch_large_pool_with_output_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_u8_maxpool_config();
|
|
ASSERT_NE(maxpool_config, nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestU8();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.output_pixel_stride(5 * channels)
|
|
.TestU8();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, setup_increasing_batch) {
|
|
ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.next_batch_size(5)
|
|
.input_height(8)
|
|
.input_width(8)
|
|
.pooling_height(5)
|
|
.pooling_width(3)
|
|
.channels(24)
|
|
.TestSetupU8();
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, setup_decreasing_batch) {
|
|
ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(5)
|
|
.next_batch_size(3)
|
|
.input_height(8)
|
|
.input_width(8)
|
|
.pooling_height(5)
|
|
.pooling_width(3)
|
|
.channels(24)
|
|
.TestSetupU8();
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, setup_changing_height) {
|
|
ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(8)
|
|
.input_width(8)
|
|
.next_input_height(9)
|
|
.pooling_height(5)
|
|
.pooling_width(3)
|
|
.channels(24)
|
|
.TestSetupU8();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(8)
|
|
.input_width(8)
|
|
.next_input_height(7)
|
|
.pooling_height(5)
|
|
.pooling_width(3)
|
|
.channels(24)
|
|
.TestSetupU8();
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, setup_changing_width) {
|
|
ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(8)
|
|
.input_width(8)
|
|
.next_input_width(9)
|
|
.pooling_height(5)
|
|
.pooling_width(3)
|
|
.channels(24)
|
|
.TestSetupU8();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(8)
|
|
.input_width(8)
|
|
.next_input_width(7)
|
|
.pooling_height(5)
|
|
.pooling_width(3)
|
|
.channels(24)
|
|
.TestSetupU8();
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_U8, setup_swap_height_and_width) {
|
|
ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestSetupU8();
|
|
}
|
|
|
|
|
|
#ifndef XNN_EXCLUDE_F16_TESTS
|
|
TEST(MAX_POOLING_NHWC_F16, unit_batch_small_1xM_pool) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, unit_batch_small_1xM_pool_with_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 3; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
for (size_t padding_left = 0; padding_left <= 1; padding_left++) {
|
|
for (size_t padding_right = 0; padding_right <= 1; padding_right++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, unit_batch_small_1xM_pool_with_tf_same_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 3; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
for (size_t input_width = pool_size; input_width <= pool_size * 2; input_width++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(input_width)
|
|
.padding_tf_same(true)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, unit_batch_small_1xM_pool_with_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 4)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.stride_width(2)
|
|
.channels(channels)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, unit_batch_small_1xM_pool_with_dilation) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(2 * pool_size + 1)
|
|
.padding_left(1)
|
|
.padding_right(1)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.dilation_width(2)
|
|
.channels(channels)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, unit_batch_small_Mx1_pool) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, unit_batch_small_Mx1_pool_with_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
for (size_t padding_top = 0; padding_top <= 1; padding_top++) {
|
|
for (size_t padding_bottom = 0; padding_bottom <= 1; padding_bottom++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, unit_batch_small_Mx1_pool_with_tf_same_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
for (size_t input_height = pool_size; input_height <= pool_size * 2; input_height++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(input_height)
|
|
.input_width(3)
|
|
.padding_tf_same(true)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, unit_batch_small_Mx1_pool_with_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 3)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.stride_height(2)
|
|
.channels(channels)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, unit_batch_small_Mx1_pool_with_dilation) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2 * pool_size)
|
|
.input_width(3)
|
|
.padding_top(1)
|
|
.padding_bottom(1)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.dilation_height(2)
|
|
.channels(channels)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, unit_batch_small_pool_with_input_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestF16();
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, unit_batch_small_pool_with_output_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestF16();
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, unit_batch_small_pool_with_qmin) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.qmin(192)
|
|
.TestF16();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.qmin(192)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, unit_batch_small_pool_with_qmax) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.qmax(192)
|
|
.TestF16();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.qmax(192)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, unit_batch_large_1xM_pool) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, unit_batch_large_1xM_pool_with_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
for (size_t padding_left = 0; padding_left <= 1; padding_left++) {
|
|
for (size_t padding_right = 0; padding_right <= 1; padding_right++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, unit_batch_large_1xM_pool_with_tf_same_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
for (size_t input_width = pool_size; input_width <= pool_size * 2; input_width++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(input_width)
|
|
.padding_tf_same(true)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, unit_batch_large_1xM_pool_with_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 4)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.stride_width(2)
|
|
.channels(channels)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, unit_batch_large_1xM_pool_with_dilation) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(2 * pool_size + 1)
|
|
.padding_left(1)
|
|
.padding_right(1)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.dilation_width(2)
|
|
.channels(channels)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, unit_batch_large_Mx1_pool) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, unit_batch_large_Mx1_pool_with_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
for (size_t padding_top = 0; padding_top <= 1; padding_top++) {
|
|
for (size_t padding_bottom = 0; padding_bottom <= 1; padding_bottom++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, unit_batch_large_Mx1_pool_with_tf_same_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
for (size_t input_height = pool_size; input_height <= pool_size * 2; input_height++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(input_height)
|
|
.input_width(3)
|
|
.padding_tf_same(true)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, unit_batch_large_Mx1_pool_with_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 3)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.stride_height(2)
|
|
.channels(channels)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, unit_batch_large_Mx1_pool_with_dilation) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2 * pool_size)
|
|
.input_width(3)
|
|
.padding_top(1)
|
|
.padding_bottom(1)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.dilation_height(2)
|
|
.channels(channels)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, unit_batch_large_pool_with_input_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestF16();
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, unit_batch_large_pool_with_output_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestF16();
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, unit_batch_large_pool_with_qmin) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.qmin(192)
|
|
.TestF16();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.qmin(192)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, unit_batch_large_pool_with_qmax) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.qmax(192)
|
|
.TestF16();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.qmax(192)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, small_batch_small_pool) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.TestF16();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, small_batch_small_pool_with_input_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestF16();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.input_pixel_stride(5 * channels)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, small_batch_small_pool_with_output_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestF16();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.output_pixel_stride(5 * channels)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, small_batch_large_pool) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.TestF16();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, small_batch_large_pool_with_input_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestF16();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.input_pixel_stride(5 * channels)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, small_batch_large_pool_with_output_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f16_maxpool_config();
|
|
if (maxpool_config == nullptr) {
|
|
GTEST_SKIP(); // F16 unsupported.
|
|
}
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestF16();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.output_pixel_stride(5 * channels)
|
|
.TestF16();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, setup_increasing_batch) {
|
|
ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.next_batch_size(5)
|
|
.input_height(8)
|
|
.input_width(8)
|
|
.pooling_height(5)
|
|
.pooling_width(3)
|
|
.channels(24)
|
|
.TestSetupF16();
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, setup_decreasing_batch) {
|
|
ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(5)
|
|
.next_batch_size(3)
|
|
.input_height(8)
|
|
.input_width(8)
|
|
.pooling_height(5)
|
|
.pooling_width(3)
|
|
.channels(24)
|
|
.TestSetupF16();
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, setup_changing_height) {
|
|
ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(8)
|
|
.input_width(8)
|
|
.next_input_height(9)
|
|
.pooling_height(5)
|
|
.pooling_width(3)
|
|
.channels(24)
|
|
.TestSetupF16();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(8)
|
|
.input_width(8)
|
|
.next_input_height(7)
|
|
.pooling_height(5)
|
|
.pooling_width(3)
|
|
.channels(24)
|
|
.TestSetupF16();
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, setup_changing_width) {
|
|
ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(8)
|
|
.input_width(8)
|
|
.next_input_width(9)
|
|
.pooling_height(5)
|
|
.pooling_width(3)
|
|
.channels(24)
|
|
.TestSetupF16();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(8)
|
|
.input_width(8)
|
|
.next_input_width(7)
|
|
.pooling_height(5)
|
|
.pooling_width(3)
|
|
.channels(24)
|
|
.TestSetupF16();
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F16, setup_swap_height_and_width) {
|
|
ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
|
|
MaxPoolingOperatorTester()
|
|
.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)
|
|
.TestSetupF16();
|
|
}
|
|
#endif // XNN_EXCLUDE_F16_TESTS
|
|
|
|
|
|
TEST(MAX_POOLING_NHWC_F32, unit_batch_small_1xM_pool) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.TestF32();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F32, unit_batch_small_1xM_pool_with_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 3; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
for (size_t padding_left = 0; padding_left <= 1; padding_left++) {
|
|
for (size_t padding_right = 0; padding_right <= 1; padding_right++) {
|
|
MaxPoolingOperatorTester()
|
|
.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(MAX_POOLING_NHWC_F32, unit_batch_small_1xM_pool_with_tf_same_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 3; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
for (size_t input_width = pool_size; input_width <= pool_size * 2; input_width++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(input_width)
|
|
.padding_tf_same(true)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.TestF32();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F32, unit_batch_small_1xM_pool_with_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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(MAX_POOLING_NHWC_F32, unit_batch_small_1xM_pool_with_dilation) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(2 * pool_size + 1)
|
|
.padding_left(1)
|
|
.padding_right(1)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.dilation_width(2)
|
|
.channels(channels)
|
|
.TestF32();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F32, unit_batch_small_Mx1_pool) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.TestF32();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F32, unit_batch_small_Mx1_pool_with_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
for (size_t padding_top = 0; padding_top <= 1; padding_top++) {
|
|
for (size_t padding_bottom = 0; padding_bottom <= 1; padding_bottom++) {
|
|
MaxPoolingOperatorTester()
|
|
.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(MAX_POOLING_NHWC_F32, unit_batch_small_Mx1_pool_with_tf_same_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
for (size_t input_height = pool_size; input_height <= pool_size * 2; input_height++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(input_height)
|
|
.input_width(3)
|
|
.padding_tf_same(true)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.TestF32();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F32, unit_batch_small_Mx1_pool_with_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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(MAX_POOLING_NHWC_F32, unit_batch_small_Mx1_pool_with_dilation) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2 * pool_size)
|
|
.input_width(3)
|
|
.padding_top(1)
|
|
.padding_bottom(1)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.dilation_height(2)
|
|
.channels(channels)
|
|
.TestF32();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F32, unit_batch_small_pool_with_input_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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();
|
|
MaxPoolingOperatorTester()
|
|
.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(MAX_POOLING_NHWC_F32, unit_batch_small_pool_with_output_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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();
|
|
MaxPoolingOperatorTester()
|
|
.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(MAX_POOLING_NHWC_F32, unit_batch_small_pool_with_qmin) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.qmin(192)
|
|
.TestF32();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.qmin(192)
|
|
.TestF32();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F32, unit_batch_small_pool_with_qmax) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.qmax(192)
|
|
.TestF32();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.qmax(192)
|
|
.TestF32();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F32, unit_batch_large_1xM_pool) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.TestF32();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F32, unit_batch_large_1xM_pool_with_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
for (size_t padding_left = 0; padding_left <= 1; padding_left++) {
|
|
for (size_t padding_right = 0; padding_right <= 1; padding_right++) {
|
|
MaxPoolingOperatorTester()
|
|
.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(MAX_POOLING_NHWC_F32, unit_batch_large_1xM_pool_with_tf_same_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
for (size_t input_width = pool_size; input_width <= pool_size * 2; input_width++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(input_width)
|
|
.padding_tf_same(true)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.TestF32();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F32, unit_batch_large_1xM_pool_with_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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(MAX_POOLING_NHWC_F32, unit_batch_large_1xM_pool_with_dilation) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(2 * pool_size + 1)
|
|
.padding_left(1)
|
|
.padding_right(1)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.dilation_width(2)
|
|
.channels(channels)
|
|
.TestF32();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F32, unit_batch_large_Mx1_pool) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.TestF32();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F32, unit_batch_large_Mx1_pool_with_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
for (size_t padding_top = 0; padding_top <= 1; padding_top++) {
|
|
for (size_t padding_bottom = 0; padding_bottom <= 1; padding_bottom++) {
|
|
MaxPoolingOperatorTester()
|
|
.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(MAX_POOLING_NHWC_F32, unit_batch_large_Mx1_pool_with_tf_same_padding) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
for (size_t input_height = pool_size; input_height <= pool_size * 2; input_height++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(input_height)
|
|
.input_width(3)
|
|
.padding_tf_same(true)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.TestF32();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F32, unit_batch_large_Mx1_pool_with_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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(MAX_POOLING_NHWC_F32, unit_batch_large_Mx1_pool_with_dilation) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2 * pool_size)
|
|
.input_width(3)
|
|
.padding_top(1)
|
|
.padding_bottom(1)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.dilation_height(2)
|
|
.channels(channels)
|
|
.TestF32();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F32, unit_batch_large_pool_with_input_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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();
|
|
MaxPoolingOperatorTester()
|
|
.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(MAX_POOLING_NHWC_F32, unit_batch_large_pool_with_output_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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();
|
|
MaxPoolingOperatorTester()
|
|
.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(MAX_POOLING_NHWC_F32, unit_batch_large_pool_with_qmin) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.qmin(192)
|
|
.TestF32();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.qmin(192)
|
|
.TestF32();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F32, unit_batch_large_pool_with_qmax) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.qmax(192)
|
|
.TestF32();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(1)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.qmax(192)
|
|
.TestF32();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F32, small_batch_small_pool) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.TestF32();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.TestF32();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F32, small_batch_small_pool_with_input_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.input_pixel_stride(5 * channels)
|
|
.TestF32();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F32, small_batch_small_pool_with_output_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = 2; pool_size <= maxpool_config->first_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.output_pixel_stride(5 * channels)
|
|
.TestF32();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F32, small_batch_large_pool) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(pool_size + 1)
|
|
.input_width(3)
|
|
.pooling_height(pool_size)
|
|
.pooling_width(1)
|
|
.channels(channels)
|
|
.TestF32();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.TestF32();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F32, small_batch_large_pool_with_input_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.input_pixel_stride(5 * channels)
|
|
.TestF32();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F32, small_batch_large_pool_with_output_stride) {
|
|
const struct xnn_maxpool_config* maxpool_config = xnn_init_f32_maxpool_config();
|
|
assert(maxpool_config != nullptr);
|
|
for (size_t channels = 1; channels <= 100; channels += 15) {
|
|
for (size_t pool_size = maxpool_config->first_pass_tile_size + 1;
|
|
pool_size <= maxpool_config->first_pass_tile_size + maxpool_config->remainder_pass_tile_size; pool_size++) {
|
|
MaxPoolingOperatorTester()
|
|
.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();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(2)
|
|
.input_width(pool_size + 2)
|
|
.pooling_height(1)
|
|
.pooling_width(pool_size)
|
|
.channels(channels)
|
|
.output_pixel_stride(5 * channels)
|
|
.TestF32();
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F32, setup_increasing_batch) {
|
|
ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.next_batch_size(5)
|
|
.input_height(8)
|
|
.input_width(8)
|
|
.pooling_height(5)
|
|
.pooling_width(3)
|
|
.channels(24)
|
|
.TestSetupF32();
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F32, setup_decreasing_batch) {
|
|
ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(5)
|
|
.next_batch_size(3)
|
|
.input_height(8)
|
|
.input_width(8)
|
|
.pooling_height(5)
|
|
.pooling_width(3)
|
|
.channels(24)
|
|
.TestSetupF32();
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F32, setup_changing_height) {
|
|
ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(8)
|
|
.input_width(8)
|
|
.next_input_height(9)
|
|
.pooling_height(5)
|
|
.pooling_width(3)
|
|
.channels(24)
|
|
.TestSetupF32();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(8)
|
|
.input_width(8)
|
|
.next_input_height(7)
|
|
.pooling_height(5)
|
|
.pooling_width(3)
|
|
.channels(24)
|
|
.TestSetupF32();
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F32, setup_changing_width) {
|
|
ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(8)
|
|
.input_width(8)
|
|
.next_input_width(9)
|
|
.pooling_height(5)
|
|
.pooling_width(3)
|
|
.channels(24)
|
|
.TestSetupF32();
|
|
MaxPoolingOperatorTester()
|
|
.batch_size(3)
|
|
.input_height(8)
|
|
.input_width(8)
|
|
.next_input_width(7)
|
|
.pooling_height(5)
|
|
.pooling_width(3)
|
|
.channels(24)
|
|
.TestSetupF32();
|
|
}
|
|
|
|
TEST(MAX_POOLING_NHWC_F32, setup_swap_height_and_width) {
|
|
ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
|
|
MaxPoolingOperatorTester()
|
|
.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();
|
|
}
|