/*
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 * All rights reserved.
 *
 * This source code is licensed under the BSD-style license found in the
 * LICENSE file in the root directory of this source tree.
 */

#include <executorch/kernels/test/FunctionHeaderWrapper.h> // Declares the operator
#include <executorch/kernels/test/TestUtil.h>
#include <executorch/kernels/test/supported_features.h>
#include <executorch/runtime/core/exec_aten/exec_aten.h>
#include <executorch/runtime/core/exec_aten/testing_util/tensor_factory.h>
#include <executorch/runtime/core/exec_aten/testing_util/tensor_util.h>

#include <gtest/gtest.h>

using namespace ::testing;

class OpAvgPool2DOutTest : public OperatorTest {
 protected:
  exec_aten::Tensor& op_avg_pool2d_out(
      const exec_aten::Tensor& self,
      exec_aten::ArrayRef<int64_t> kernel_size,
      exec_aten::ArrayRef<int64_t> stride,
      exec_aten::ArrayRef<int64_t> padding,
      bool ceil_mode,
      bool count_include_pad,
      exec_aten::optional<int64_t> divisor_override,
      exec_aten::Tensor& out) {
    return torch::executor::aten::avg_pool2d_outf(
        context_,
        self,
        kernel_size,
        stride,
        padding,
        ceil_mode,
        count_include_pad,
        divisor_override,
        out);
  }
};

TEST_F(OpAvgPool2DOutTest, SanityCheck4D) {
  torch::executor::testing::TensorFactory<exec_aten::ScalarType::Float> tfFloat;

  exec_aten::Tensor self = tfFloat.make(
      {2, 3, 8, 8},
      {-81.875, -37.25,  -99.75,  -60.5,   -1.375,  25.625,  -54.25,  -95.875,
       48.0,    26.125,  -14.625, -5.0,    98.25,   1.75,    5.875,   -71.25,
       -26.5,   36.0,    32.375,  -87.875, -43.125, -40.875, -49.0,   -33.0,
       16.0,    -1.875,  -48.0,   77.0,    87.0,    58.25,   20.0,    10.875,
       -37.875, -88.25,  97.75,   -98.25,  -52.0,   -92.75,  -89.375, -31.5,
       -1.625,  -50.0,   -6.625,  -62.5,   -86.5,   -96.5,   85.0,    -94.375,
       -3.625,  94.125,  12.25,   -33.875, -25.625, -20.625, -56.5,   -78.5,
       8.25,    38.875,  74.5,    -22.75,  14.125,  46.5,    -28.625, -52.0,
       -36.75,  -3.75,   48.875,  2.875,   76.125,  -42.125, -71.125, 22.5,
       9.25,    98.5,    41.5,    65.625,  -82.25,  -85.75,  -20.75,  82.375,
       3.25,    -74.5,   -14.25,  18.75,   60.25,   -21.875, 7.0,     -44.0,
       -97.0,   -71.625, 65.75,   -89.5,   26.375,  14.125,  -99.75,  42.625,
       -11.25,  56.625,  -63.0,   -34.25,  -76.125, 69.25,   22.875,  43.875,
       -7.125,  -16.875, 93.875,  -48.25,  -57.375, -30.625, 11.75,   73.0,
       -96.5,   14.5,    50.375,  -65.875, 73.625,  88.625,  -11.375, 41.75,
       7.5,     62.125,  -1.25,   81.125,  -78.375, 62.875,  75.125,  -34.75,
       50.5,    -2.125,  64.125,  -74.375, 22.875,  39.625,  -91.125, 96.125,
       -31.75,  -70.875, 12.75,   6.5,     30.125,  12.75,   -77.375, -56.5,
       -24.5,   -59.25,  88.125,  -26.375, -46.0,   62.75,   39.25,   -78.375,
       67.875,  -57.75,  15.5,    80.5,    -70.5,   96.875,  -18.625, -53.625,
       70.75,   70.375,  -68.125, -41.0,   -59.875, 74.625,  -82.125, 35.125,
       -24.25,  33.25,   45.875,  -74.75,  37.0,    -0.875,  86.625,  -14.375,
       45.25,   41.0,    -10.625, -49.875, 61.5,    -10.375, -39.0,   75.5,
       -72.625, -29.0,   -53.875, -31.375, -57.375, 79.5,    -25.0,   -83.0,
       -49.875, -74.875, 50.25,   8.25,    -35.5,   -0.875,  27.125,  85.875,
       -95.625, 17.875,  45.25,   -56.75,  60.5,    -61.75,  47.625,  50.5,
       83.0,    -63.25,  40.375,  -61.75,  -89.0,   -66.5,   6.0,     46.0,
       -20.625, -15.625, -58.0,   88.25,   31.25,   -37.5,   -38.25,  68.5,
       11.625,  34.125,  -39.5,   84.25,   27.0,    -85.625, 7.625,   86.375,
       41.625,  28.0,    -41.125, 44.625,  19.625,  -43.125, -23.875, 54.5,
       50.0,    -88.25,  29.0,    -77.5,   -82.125, -84.0,   8.75,    -32.875,
       27.75,   -88.75,  -33.25,  -58.25,  41.5,    -84.0,   -53.375, -85.5,
       -15.625, 39.75,   29.375,  -45.375, 96.5,    65.125,  34.875,  75.375,
       -32.75,  -9.75,   -55.0,   38.0,    31.125,  -35.0,   -74.375, -61.0,
       -63.75,  4.75,    88.75,   -83.25,  -19.75,  5.875,   88.375,  52.25,
       70.125,  -81.5,   -56.375, -98.375, 97.625,  88.375,  22.625,  -100.0,
       -75.0,   10.0,    41.0,    40.375,  12.0,    72.125,  31.875,  -22.25,
       -63.875, 10.5,    -81.25,  4.25,    43.5,    -44.0,   71.5,    -29.625,
       -3.5,    -91.5,   45.375,  88.875,  -93.125, -50.25,  -6.375,  -88.875,
       -2.375,  -17.75,  49.25,   -14.75,  14.75,   -2.25,   51.25,   -57.875,
       43.875,  87.0,    86.25,   -95.125, 11.75,   -26.5,   29.875,  89.25,
       -18.5,   55.375,  74.25,   -64.0,   51.875,  78.0,    82.625,  -34.0,
       -0.875,  -69.375, -90.875, -83.5,   13.625,  46.875,  8.375,   16.875,
       96.75,   31.25,   45.625,  -2.625,  -71.0,   -62.375, 31.25,   -23.5,
       66.0,    51.375,  -45.25,  43.375,  49.5,    12.625,  -73.875, 26.375,
       29.0,    -86.5,   -55.375, 88.75,   20.0,    90.0,    28.75,   -12.875,
       -37.75,  -1.875,  -28.125, 96.75,   -66.75,  48.375,  -79.25,  8.0,
       -14.25,  -8.0,    51.75,   28.375,  32.0,    -50.875, 53.0,    -81.75});
  ::std::vector<int64_t> kernel_size_vec = {2, 3};
  exec_aten::ArrayRef<int64_t> kernel_size = exec_aten::ArrayRef<int64_t>(
      kernel_size_vec.data(), kernel_size_vec.size());
  ::std::vector<int64_t> stride_vec = {3, 2};
  exec_aten::ArrayRef<int64_t> stride =
      exec_aten::ArrayRef<int64_t>(stride_vec.data(), stride_vec.size());
  ::std::vector<int64_t> padding_vec = {1, 1};
  exec_aten::ArrayRef<int64_t> padding =
      exec_aten::ArrayRef<int64_t>(padding_vec.data(), padding_vec.size());
  bool ceil_mode = false;
  bool count_include_pad = true;
  exec_aten::optional<int64_t> divisor_override;
  exec_aten::Tensor out = tfFloat.zeros({2, 3, 3, 4});
  exec_aten::Tensor out_expected = tfFloat.make(
      {2, 3, 3, 4},
      {-19.85416603088379,
       -32.91666793823242,
       -6.041666507720947,
       -20.75,
       3.9375,
       1.2708333730697632,
       8.395833015441895,
       -5.625,
       6.479166507720947,
       -7.770833492279053,
       -54.27083206176758,
       -43.58333206176758,
       -6.75,
       8.0,
       6.145833492279053,
       -15.125,
       -39.97916793823242,
       -27.5625,
       1.3541666269302368,
       -16.97916603088379,
       -17.66666603088379,
       4.625,
       -6.645833492279053,
       28.85416603088379,
       8.0625,
       -2.0625,
       -1.9791666269302368,
       7.4375,
       -12.270833015441895,
       6.791666507720947,
       16.20833396911621,
       8.041666984558105,
       15.875,
       -2.5208332538604736,
       -6.229166507720947,
       16.25,
       -20.79166603088379,
       -2.7291667461395264,
       -4.6875,
       18.6875,
       -2.75,
       -11.666666984558105,
       -22.54166603088379,
       -3.625,
       5.229166507720947,
       -17.54166603088379,
       -37.08333206176758,
       -20.10416603088379,
       4.020833492279053,
       3.9583332538604736,
       19.375,
       29.22916603088379,
       -11.729166984558105,
       -37.66666793823242,
       -1.5833333730697632,
       26.25,
       -24.72916603088379,
       -3.9583332538604736,
       -8.458333015441895,
       -24.60416603088379,
       21.8125,
       13.020833015441895,
       -18.3125,
       15.4375,
       9.625,
       -28.25,
       -26.5,
       2.9166667461395264,
       -16.1875,
       2.2708332538604736,
       46.1875,
       13.833333015441895});
  op_avg_pool2d_out(
      self,
      kernel_size,
      stride,
      padding,
      ceil_mode,
      count_include_pad,
      divisor_override,
      out);
  EXPECT_TENSOR_CLOSE(out, out_expected);
}

TEST_F(OpAvgPool2DOutTest, SanityCheck4DDivisorOverride) {
  torch::executor::testing::TensorFactory<exec_aten::ScalarType::Float> tfFloat;

  exec_aten::Tensor self = tfFloat.make(
      {2, 3, 8, 8},
      {13.25,   87.125,  -61.875, 52.875,  -74.5,   37.5,    -62.125, 25.375,
       -34.375, 68.5,    87.875,  91.125,  -22.75,  96.875,  79.25,   38.125,
       6.5,     27.625,  58.875,  41.5,    -82.875, -77.875, -3.25,   22.0,
       -73.375, -56.875, 56.25,   -6.375,  -16.375, 40.0,    22.625,  61.0,
       -78.375, -87.5,   23.25,   -55.625, 92.75,   -90.0,   -55.625, 16.25,
       -46.5,   91.75,   74.0,    -45.0,   -34.875, -46.375, 38.0,    63.375,
       22.625,  16.25,   91.125,  68.375,  98.25,   34.0,    91.25,   89.375,
       1.0,     22.875,  -93.75,  7.5,     -81.375, 1.125,   -15.875, 16.75,
       29.25,   -14.0,   77.375,  -64.625, -22.375, -97.25,  -26.75,  9.125,
       -3.5,    -64.5,   35.625,  58.25,   4.125,   85.0,    -39.125, -98.25,
       -0.625,  -29.5,   -35.375, 90.375,  37.625,  57.875,  -59.625, -81.875,
       73.5,    -37.375, 56.75,   88.25,   -52.75,  -4.75,   80.625,  -38.375,
       -55.375, -76.625, -59.875, -41.5,   76.875,  45.375,  24.625,  57.5,
       58.25,   64.375,  -38.125, -39.875, 71.125,  2.5,     -14.75,  21.75,
       -8.375,  5.0,     71.625,  29.875,  79.375,  -13.125, -22.625, 96.875,
       7.5,     64.5,    1.0,     -40.75,  -56.0,   -94.75,  -89.5,   -71.25,
       78.125,  -5.875,  93.125,  -20.125, -73.875, -34.625, 51.5,    46.5,
       -88.25,  3.25,    26.0,    -32.375, -77.625, -51.875, 0.75,    85.75,
       -15.875, 84.625,  -44.75,  -87.625, 17.25,   -59.25,  40.25,   -0.5,
       14.625,  -80.125, -58.125, 17.25,   -7.5,    68.875,  -99.5,   -82.75,
       75.5,    -95.25,  -63.25,  55.25,   -17.125, 64.875,  -56.25,  39.75,
       -35.25,  18.0,    21.125,  -61.125, -35.5,   12.75,   -94.125, -17.125,
       -41.375, 22.625,  -69.25,  51.375,  7.25,    14.625,  70.25,   -63.5,
       -10.0,   53.625,  -46.75,  77.5,    -78.5,   -67.25,  -53.375, -33.5,
       23.375,  77.625,  64.125,  -56.375, -59.25,  -89.125, 16.25,   -2.125,
       -82.125, 72.75,   45.5,    -36.375, -1.125,  -7.25,   -44.5,   27.125,
       -1.75,   16.5,    71.0,    81.25,   4.125,   74.0,    45.0,    -4.75,
       68.125,  30.625,  11.625,  -88.5,   48.125,  36.875,  -32.0,   9.0,
       -67.5,   -17.125, -87.5,   29.875,  23.875,  -33.0,   -6.625,  0.5,
       28.25,   54.75,   87.5,    28.0,    -48.5,   -98.0,   -55.0,   62.25,
       -93.125, -41.5,   94.25,   -28.375, -23.75,  -50.625, 90.375,  -91.0,
       -67.75,  -11.625, 24.5,    49.5,    21.0,    70.25,   66.0,    69.75,
       43.25,   -14.875, -93.875, -15.75,  94.5,    69.125,  -67.625, 15.875,
       -26.875, 41.625,  -64.375, 23.0,    13.375,  78.5,    -88.875, 17.625,
       -52.375, 55.25,   41.875,  62.625,  29.0,    53.875,  -93.625, -93.375,
       -70.375, -56.125, 9.0,     60.625,  4.375,   66.625,  -46.25,  43.125,
       93.625,  85.125,  -91.375, 49.375,  40.25,   -19.75,  13.0,    69.5,
       -42.375, 47.25,   7.0,     -93.625, -80.0,   -57.875, -94.875, 9.375,
       -2.5,    -42.25,  -8.0,    92.5,    6.875,   -11.0,   80.625,  27.125,
       -75.875, -57.375, -36.625, 17.375,  33.125,  -49.0,   -84.0,   -45.5,
       -9.625,  -16.125, 8.75,    -1.0,    -16.625, 2.5,     -56.0,   -52.0,
       85.0,    63.5,    -50.625, 7.375,   -90.0,   -75.875, -82.0,   82.75,
       16.125,  -35.25,  -42.875, -1.375,  39.875,  -39.625, -66.875, 29.875,
       -80.75,  66.5,    -50.125, 13.5,    -77.875, 6.125,   -91.625, -44.125,
       -50.125, 59.625,  -32.5,   -2.0,    -83.375, 21.0,    49.875,  48.125,
       -93.375, -54.875, -61.125, 96.375,  91.25,   -2.375,  -33.25,  48.125,
       -58.125, -50.75,  -50.875, 8.375,   35.625,  -72.5,   -76.125, -33.25,
       -18.75,  -71.0,   76.625,  -11.25,  -3.0,    -38.625, -66.375, -25.0});
  ::std::vector<int64_t> kernel_size_vec = {2, 3};
  exec_aten::ArrayRef<int64_t> kernel_size = exec_aten::ArrayRef<int64_t>(
      kernel_size_vec.data(), kernel_size_vec.size());
  ::std::vector<int64_t> stride_vec = {3, 2};
  exec_aten::ArrayRef<int64_t> stride =
      exec_aten::ArrayRef<int64_t>(stride_vec.data(), stride_vec.size());
  ::std::vector<int64_t> padding_vec = {1, 1};
  exec_aten::ArrayRef<int64_t> padding =
      exec_aten::ArrayRef<int64_t>(padding_vec.data(), padding_vec.size());
  bool ceil_mode = false;
  bool count_include_pad = true;
  exec_aten::optional<int64_t> divisor_override =
      exec_aten::optional<int64_t>(10);
  exec_aten::Tensor out = tfFloat.zeros({2, 3, 3, 4});
  exec_aten::Tensor out_expected = tfFloat.make(
      {2, 3, 3, 4},
      {10.037500381469727,
       7.8125,
       1.587499976158142,
       0.07500000298023224,
       -9.612500190734863,
       12.100000381469727,
       -10.199999809265137,
       6.449999809265137,
       8.412500381469727,
       29.649999618530273,
       7.4375,
       26.962499618530273,
       1.524999976158142,
       -0.125,
       -18.424999237060547,
       -11.487500190734863,
       0.6000000238418579,
       13.3125,
       21.662500381469727,
       -4.612500190734863,
       11.925000190734863,
       9.287500381469727,
       12.987500190734863,
       7.0625,
       7.224999904632568,
       6.712500095367432,
       -12.862500190734863,
       6.337500095367432,
       0.32499998807907104,
       -16.875,
       -5.099999904632568,
       -13.287500381469727,
       -3.5999999046325684,
       -1.725000023841858,
       -1.0625,
       -7.712500095367432,
       10.100000381469727,
       8.537500381469727,
       -20.475000381469727,
       -7.5,
       11.350000381469727,
       12.25,
       15.587499618530273,
       12.8125,
       -5.162499904632568,
       19.462499618530273,
       -22.125,
       -14.199999809265137,
       2.8375000953674316,
       -12.449999809265137,
       14.787500381469727,
       1.7374999523162842,
       -12.362500190734863,
       17.325000762939453,
       27.712499618530273,
       -6.962500095367432,
       -3.987499952316284,
       0.2874999940395355,
       -14.3125,
       -4.662499904632568,
       -2.575000047683716,
       -0.8374999761581421,
       -1.5125000476837158,
       -10.550000190734863,
       -3.3375000953674316,
       -4.962500095367432,
       -5.9375,
       -20.625,
       -25.712499618530273,
       -11.287500381469727,
       15.675000190734863,
       -16.9375});
  op_avg_pool2d_out(
      self,
      kernel_size,
      stride,
      padding,
      ceil_mode,
      count_include_pad,
      divisor_override,
      out);
  EXPECT_TENSOR_CLOSE(out, out_expected);
}

TEST_F(OpAvgPool2DOutTest, SanityCheck4DCeilModeNoIncludePadding) {
  torch::executor::testing::TensorFactory<exec_aten::ScalarType::Float> tfFloat;

  exec_aten::Tensor self = tfFloat.make(
      {2, 3, 14, 12},
      {26.375,  -17.0,   63.5,    83.0,    21.375,  -46.5,   -69.125, 99.875,
       -67.125, -76.0,   -1.125,  -2.625,  -48.0,   -1.5,    -92.75,  95.5,
       91.5,    32.625,  -35.25,  12.125,  55.0,    -64.875, -53.125, -20.0,
       -5.875,  -0.25,   -3.5,    4.875,   81.625,  92.25,   98.25,   -47.375,
       78.875,  -97.5,   10.375,  55.75,   65.375,  97.375,  -93.625, 51.5,
       62.75,   -37.625, 1.875,   -73.125, 78.75,   -62.25,  20.625,  89.375,
       16.75,   43.0,    98.375,  17.0,    40.125,  -78.0,   -24.375, 9.875,
       2.0,     -57.125, -65.375, -86.0,   -98.375, 68.75,   70.375,  16.25,
       -24.625, 12.0,    -55.0,   -17.0,   81.375,  12.0,    62.875,  -79.0,
       46.0,    -12.625, -42.0,   3.125,   -6.0,    88.5,    -42.25,  82.625,
       -96.875, 51.375,  41.875,  36.125,  -27.375, 33.375,  -81.75,  6.875,
       -39.5,   -57.75,  24.0,    35.375,  -22.875, 96.375,  -65.0,   -67.0,
       -64.875, 86.625,  70.625,  16.5,    -45.0,   55.375,  88.5,    -42.0,
       -38.5,   -71.5,   -66.125, -30.5,   59.375,  -8.25,   8.375,   85.5,
       20.625,  -58.875, -35.5,   -80.125, 93.25,   -15.5,   40.125,  54.25,
       -40.25,  85.125,  61.375,  84.25,   58.5,    -60.375, -58.375, -81.5,
       -80.125, -84.0,   31.75,   55.125,  8.125,   -56.625, -7.0,    12.625,
       75.75,   -88.125, -5.375,  60.0,    21.25,   62.125,  2.125,   -3.375,
       -26.75,  44.25,   -96.5,   -87.5,   -36.625, -36.125, 1.125,   -42.0,
       -48.125, 24.0,    80.375,  -38.5,   50.75,   -0.625,  35.625,  31.875,
       -32.75,  70.375,  -41.0,   -20.75,  27.5,    60.75,   -10.5,   54.875,
       68.25,   40.0,    0.875,   39.375,  67.375,  31.25,   79.25,   37.625,
       33.0,    10.625,  -47.5,   -21.0,   -2.625,  81.0,    -42.25,  23.625,
       10.375,  1.625,   11.0,    14.625,  19.5,    -36.75,  99.875,  86.125,
       -70.25,  26.375,  -86.375, -21.0,   -25.0,   51.75,   -11.375, 58.875,
       74.75,   77.625,  90.0,    -40.5,   -32.625, 40.5,    33.0,    -80.625,
       -33.875, 38.5,    30.625,  -55.375, 53.875,  -1.25,   75.0,    -55.375,
       71.125,  -34.625, -29.75,  -92.75,  -34.125, -17.25,  -72.0,   79.5,
       81.5,    -14.375, -35.625, -38.875, 97.625,  -3.0,    -31.125, 31.25,
       13.5,    -3.625,  -64.0,   72.0,    64.5,    6.875,   -4.875,  19.875,
       50.25,   80.75,   76.875,  91.75,   -26.375, 94.125,  19.0,    4.375,
       -48.625, -69.375, 83.0,    6.75,    -85.5,   -64.75,  -99.875, -88.375,
       36.875,  24.375,  96.0,    24.25,   -10.0,   39.75,   -8.75,   -76.25,
       -20.75,  -96.25,  75.5,    23.625,  50.875,  76.375,  -74.25,  2.0,
       57.125,  31.875,  -67.375, 78.375,  -25.125, -27.75,  -82.375, 37.25,
       84.875,  8.25,    53.75,   -14.375, -68.875, -7.875,  -5.25,   50.25,
       -92.75,  -90.625, 64.0,    -32.0,   -20.75,  47.625,  35.5,    -17.875,
       44.0,    35.625,  -38.0,   -80.5,   74.375,  -11.125, 85.625,  47.25,
       -92.375, 2.375,   -5.375,  31.875,  -83.5,   -41.0,   -72.25,  49.75,
       72.125,  -92.875, -39.125, -19.75,  -79.25,  90.375,  -67.625, 71.375,
       -9.75,   -80.125, -56.875, 27.0,    93.75,   55.75,   85.0,    -60.5,
       -11.625, 11.5,    66.0,    7.875,   72.125,  -15.25,  60.25,   58.0,
       -16.125, -41.875, -16.875, 32.25,   -69.75,  81.5,    -65.125, -28.0,
       59.25,   2.375,   -8.25,   60.0,    -69.625, 3.0,     59.875,  93.25,
       41.875,  -38.625, -86.375, -18.25,  86.375,  29.875,  5.375,   -63.0,
       -40.875, 42.0,    51.125,  -35.125, 3.125,   31.875,  29.75,   6.5,
       -12.125, -86.875, 62.5,    6.75,    -43.125, 37.0,    58.625,  -89.5,
       -62.125, -21.0,   -95.875, -42.125, -42.125, -52.125, -22.875, 78.375,
       -53.375, -85.625, 19.375,  -28.0,   47.375,  63.875,  21.75,   85.75,
       74.375,  27.75,   46.0,    41.375,  55.125,  -4.375,  -59.375, -93.375,
       -95.375, 98.125,  -30.875, 69.25,   69.75,   -78.375, 71.125,  -71.25,
       -73.75,  69.0,    -68.125, -49.0,   50.0,    51.25,   -9.625,  -4.625,
       -85.75,  -99.125, -99.875, -65.5,   28.75,   95.25,   79.625,  16.5,
       37.125,  -62.75,  -62.0,   65.25,   95.5,    -15.5,   -42.375, 35.75,
       18.5,    89.25,   35.5,    -90.375, 3.625,   -2.875,  55.25,   63.5,
       1.25,    2.625,   -18.375, -69.0,   9.5,     -31.375, -44.375, 35.125,
       -37.0,   20.875,  98.5,    -43.0,   -35.625, -8.125,  40.25,   -89.125,
       41.5,    -100.0,  -39.625, -76.5,   39.0,    6.125,   22.75,   83.75,
       -62.375, 31.75,   -86.625, -26.125, 87.25,   -69.0,   47.0,    99.125,
       73.5,    24.875,  84.875,  13.625,  -57.875, 67.125,  -10.375, -67.75,
       -92.0,   52.125,  66.875,  -62.25,  61.125,  -46.875, -35.875, 35.625,
       -91.5,   -95.25,  11.625,  51.125,  38.875,  -1.875,  1.75,    11.125,
       61.125,  -58.25,  60.125,  -62.25,  -83.75,  -41.375, -14.75,  48.25,
       -83.25,  -91.75,  13.625,  -12.75,  30.25,   -54.125, 60.0,    -25.875,
       -45.0,   -33.75,  85.875,  46.0,    -74.75,  -93.375, 64.25,   -31.5,
       -30.125, 85.25,   87.0,    -1.75,   -55.5,   32.0,    -67.25,  3.125,
       -66.0,   16.75,   -62.25,  -59.375, 91.375,  -49.625, -74.125, 52.5,
       -9.125,  70.5,    69.375,  -86.25,  -67.75,  82.625,  37.5,    1.5,
       73.125,  -68.875, -1.5,    42.875,  94.125,  98.125,  9.5,     99.375,
       35.0,    -17.625, -22.375, 75.0,    67.5,    47.25,   -12.625, 30.375,
       -16.625, 97.125,  86.5,    61.375,  -94.875, 99.125,  62.25,   -83.875,
       -65.0,   69.125,  84.125,  -84.75,  90.25,   -77.0,   29.625,  58.375,
       8.625,   76.875,  67.375,  -89.875, 62.5,    51.0,    77.75,   -0.375,
       45.5,    86.875,  -49.5,   -71.375, -32.75,  -76.375, -29.375, 69.25,
       -13.375, 95.875,  -58.375, 12.75,   95.375,  83.375,  51.0,    40.75,
       58.125,  -26.875, 29.375,  81.625,  37.25,   -91.375, 1.25,    20.625,
       -46.5,   24.875,  12.25,   -73.25,  -1.625,  -69.625, -79.0,   60.125,
       -0.125,  41.875,  -24.375, 66.5,    9.25,    -72.625, -66.375, -29.5,
       -51.5,   28.75,   -79.75,  16.125,  -55.5,   80.25,   -14.25,  37.875,
       98.625,  -6.875,  67.75,   90.875,  -80.375, 71.625,  58.875,  -63.0,
       -71.625, 33.5,    -30.125, -33.25,  94.75,   -30.5,   -45.75,  62.875,
       67.75,   -22.125, 22.0,    48.75,   -35.75,  83.625,  -53.375, 55.875,
       53.875,  38.0,    0.375,   15.125,  -29.25,  55.0,    -79.875, 40.5,
       65.25,   -67.625, -43.125, -47.625, -81.375, 40.5,    -94.375, 26.0,
       70.0,    50.25,   53.0,    26.75,   0.875,   4.0,     17.625,  -74.25,
       -28.5,   56.5,    71.25,   -67.0,   -9.875,  15.25,   -38.125, -61.375,
       -22.25,  14.625,  -47.25,  -7.125,  34.75,   10.625,  -10.25,  0.5,
       77.5,    36.0,    -51.875, -59.5,   13.375,  45.75,   -71.625, -29.75,
       -90.75,  -38.25,  52.0,    77.875,  21.125,  56.625,  17.875,  -32.625,
       -39.0,   -68.75,  -82.875, -1.25,   96.75,   -91.375, -22.5,   27.5,
       -30.125, -23.875, -76.0,   -93.5,   37.5,    78.375,  -12.5,   -84.875,
       -21.5,   29.25,   -4.375,  -48.875, 90.75,   -77.375, -23.625, -5.75,
       -75.125, -92.125, 9.0,     96.5,    76.125,  47.5,    15.875,  21.0,
       18.625,  94.25,   8.5,     -67.75,  69.25,   -72.0,   -14.5,   90.5,
       -31.875, -20.875, -93.625, 42.875,  -54.375, -70.5,   -75.375, -6.75,
       -48.875, 21.75,   1.75,    45.5,    60.75,   -9.75,   13.625,  -87.125,
       12.625,  71.125,  -46.5,   -31.25,  -15.375, -42.75,  13.0,    26.25,
       61.625,  81.0,    18.125,  84.5,    31.75,   -14.125, -80.875, 18.875,
       51.125,  0.625,   41.5,    -81.625, 32.625,  -19.375, 76.875,  54.0,
       -40.375, -78.25,  48.0,    77.5,    21.0,    71.25,   34.875,  73.25,
       71.75,   87.25,   -89.375, 93.625,  64.25,   -81.375, -15.0,   -58.125,
       64.75,   -49.125, 72.125,  -62.0,   0.125,   -56.0,   -6.875,  98.25,
       -24.75,  48.375,  -85.25,  -62.25,  98.25,   12.75,   -67.25,  72.125,
       9.625,   8.5,     -43.375, 15.25,   -50.875, 7.875,   -21.625, 97.25,
       48.5,    33.25,   -15.125, 41.125,  69.25,   -35.875, 64.5,    98.75,
       -78.125, 44.625,  75.75,   41.625,  -39.375, -4.875,  87.5,    -67.25,
       -23.25,  -52.875, -44.875, -70.75,  -63.75,  86.25,   -44.0,   -92.375,
       -22.75,  -70.75,  -84.875, -98.125, -86.75,  35.25,   -77.5,   -78.0,
       -62.5,   -88.5,   58.25,   -65.125, 70.25,   -5.875,  11.25,   47.125,
       10.25,   -91.75,  -50.125, 45.125,  49.625,  57.25,   26.375,  -74.75,
       70.25,   38.625,  51.25,   12.375,  -79.875, -76.25,  12.0,    -15.625,
       50.25,   72.125,  30.5,    -68.625, -22.875, 5.125,   -59.5,   -82.75,
       -64.5,   46.75,   -87.75,  17.875,  50.875,  69.625,  24.125,  -25.0,
       -25.0,   -5.0,    21.0,    51.875,  91.625,  50.0,    77.375,  -51.0,
       64.25,   41.0,    -20.875, -89.75,  -74.75,  48.375,  -41.75,  99.25,
       36.125,  10.75,   -49.625, 57.375,  19.625,  -43.75,  16.625,  -48.375,
       84.375,  31.125,  93.375,  76.125,  -12.25,  98.5,    15.0,    71.625,
       -87.75,  98.625,  -74.5,   -42.75,  -73.5,   -30.875, -44.625, 70.5,
       96.0,    21.5,    48.375,  12.75,   -65.625, 56.5,    97.375,  29.5,
       -19.375, 94.625,  11.25,   -0.375,  -96.5,   -48.75,  -40.375, 98.0,
       -43.0,   20.75,   50.375,  97.875,  -72.875, 92.125,  98.5,    -74.0,
       -57.875, 56.375,  -34.0,   6.125,   -66.125, 71.75,   -67.125, -79.0,
       -85.875, -88.625, 97.375,  16.875,  -18.5,   -75.125, 22.625,  -21.125,
       3.875,   57.25,   97.75,   -11.0,   -46.875, 16.75,   -69.25,  99.75,
       -68.25,  99.75,   -17.25,  -3.125,  34.25,   -54.125, -93.125, 65.0,
       -76.375, -20.625, -77.875, -65.625, -79.875, 28.75,   58.25,   -25.25});
  ::std::vector<int64_t> kernel_size_vec = {4, 2};
  exec_aten::ArrayRef<int64_t> kernel_size = exec_aten::ArrayRef<int64_t>(
      kernel_size_vec.data(), kernel_size_vec.size());
  ::std::vector<int64_t> stride_vec = {1, 2};
  exec_aten::ArrayRef<int64_t> stride =
      exec_aten::ArrayRef<int64_t>(stride_vec.data(), stride_vec.size());
  ::std::vector<int64_t> padding_vec = {1, 1};
  exec_aten::ArrayRef<int64_t> padding =
      exec_aten::ArrayRef<int64_t>(padding_vec.data(), padding_vec.size());
  bool ceil_mode = true;
  bool count_include_pad = false;
  exec_aten::optional<int64_t> divisor_override;
  exec_aten::Tensor out = tfFloat.zeros({2, 3, 13, 7});
  exec_aten::Tensor out_expected = tfFloat.make(
      {2, 3, 13, 7},
      {-9.166666984558105,
       -8.583333015441895,
       62.97916793823242,
       12.041666984558105,
       21.89583396911621,
       -47.04166793823242,
       11.041666984558105,
       9.46875,
       -5.96875,
       61.515625,
       4.5625,
       17.125,
       -40.484375,
       30.625,
       7.0625,
       5.890625,
       55.609375,
       6.21875,
       14.515625,
       -46.15625,
       9.78125,
       -5.53125,
       35.0625,
       31.1875,
       1.171875,
       14.171875,
       -22.046875,
       -4.96875,
       7.4375,
       28.703125,
       20.015625,
       -16.859375,
       8.453125,
       0.5,
       -9.875,
       -15.75,
       22.1875,
       1.65625,
       -16.609375,
       9.3125,
       9.625,
       -48.96875,
       -36.15625,
       24.171875,
       -9.046875,
       14.171875,
       -2.234375,
       7.734375,
       -35.09375,
       3.28125,
       6.796875,
       5.265625,
       7.75,
       -8.640625,
       1.453125,
       -1.78125,
       -18.28125,
       31.9375,
       23.46875,
       -12.875,
       -27.0625,
       -16.734375,
       2.96875,
       -9.40625,
       30.03125,
       38.59375,
       -20.34375,
       -18.46875,
       -12.625,
       18.875,
       0.125,
       3.84375,
       26.640625,
       -42.703125,
       -19.671875,
       17.625,
       16.875,
       -2.03125,
       8.203125,
       13.265625,
       -27.234375,
       -20.46875,
       20.828125,
       17.03125,
       10.708333015441895,
       -13.479166984558105,
       -6.104166507720947,
       -16.52083396911621,
       -0.3541666567325592,
       36.47916793823242,
       4.333333492279053,
       -1.5416666269302368,
       3.2708332538604736,
       15.791666984558105,
       27.25,
       39.72916793823242,
       32.3125,
       8.208333015441895,
       -9.3125,
       11.640625,
       -2.46875,
       29.078125,
       29.609375,
       33.453125,
       -7.6875,
       -8.59375,
       -1.515625,
       -31.671875,
       4.109375,
       40.90625,
       31.8125,
       -12.15625,
       16.46875,
       -10.625,
       -30.328125,
       -5.921875,
       53.703125,
       24.171875,
       -28.71875,
       46.59375,
       16.578125,
       -16.40625,
       3.171875,
       31.46875,
       4.921875,
       -16.90625,
       33.375,
       -13.1875,
       -8.53125,
       9.578125,
       33.4375,
       -0.421875,
       -22.125,
       10.40625,
       -7.734375,
       16.640625,
       21.0,
       20.703125,
       1.390625,
       7.1875,
       -20.28125,
       -17.234375,
       26.3125,
       37.203125,
       -6.765625,
       -0.5,
       14.78125,
       -56.03125,
       -40.265625,
       11.546875,
       33.453125,
       2.03125,
       -2.5,
       -7.03125,
       -16.0625,
       -10.375,
       12.34375,
       18.03125,
       -6.203125,
       -20.53125,
       24.46875,
       7.15625,
       -24.28125,
       -9.34375,
       20.609375,
       -5.890625,
       -33.21875,
       11.625,
       36.875,
       7.078125,
       -33.625,
       22.546875,
       14.515625,
       -25.953125,
       13.5625,
       80.08333587646484,
       13.875,
       -36.04166793823242,
       16.20833396911621,
       15.0,
       -34.20833206176758,
       44.91666793823242,
       -42.20833206176758,
       16.20833396911621,
       10.9375,
       -7.833333492279053,
       15.625,
       0.8333333134651184,
       1.25,
       -42.4375,
       24.109375,
       -10.75,
       -20.484375,
       1.1875,
       -8.75,
       20.53125,
       -51.75,
       23.171875,
       -3.640625,
       -11.828125,
       17.296875,
       1.203125,
       15.875,
       -20.5625,
       7.34375,
       -44.125,
       12.203125,
       26.15625,
       -4.109375,
       13.8125,
       -28.78125,
       -4.1875,
       -40.0,
       9.703125,
       15.5625,
       -25.9375,
       -4.25,
       -10.8125,
       5.71875,
       -14.34375,
       8.71875,
       46.1875,
       -23.796875,
       -14.90625,
       7.15625,
       29.59375,
       -27.609375,
       4.5625,
       34.265625,
       -34.984375,
       -42.5,
       -4.25,
       28.09375,
       -4.25,
       11.078125,
       7.0625,
       -30.0625,
       -46.96875,
       24.5625,
       10.53125,
       -9.0625,
       9.484375,
       21.03125,
       -12.046875,
       -37.125,
       39.1875,
       -14.078125,
       5.8125,
       38.796875,
       -4.59375,
       2.28125,
       -63.0,
       11.5625,
       -14.796875,
       16.515625,
       21.90625,
       -19.671875,
       -6.203125,
       -32.96875,
       18.90625,
       -5.34375,
       25.78125,
       7.21875,
       -28.09375,
       -17.234375,
       1.375,
       11.375,
       16.14583396911621,
       40.625,
       4.8125,
       -41.02083206176758,
       -13.833333015441895,
       10.541666984558105,
       -74.66666412353516,
       -25.45833396911621,
       -2.0208332538604736,
       9.0625,
       -14.125,
       26.125,
       -12.375,
       -72.9375,
       -4.078125,
       7.8125,
       -2.0,
       6.53125,
       33.046875,
       15.5625,
       -43.375,
       0.6875,
       23.4375,
       1.59375,
       17.109375,
       49.484375,
       19.40625,
       -48.40625,
       24.5,
       12.53125,
       -0.78125,
       24.953125,
       47.96875,
       33.21875,
       -29.75,
       48.21875,
       5.109375,
       30.78125,
       25.171875,
       35.15625,
       36.9375,
       -21.0,
       19.984375,
       2.765625,
       44.265625,
       21.5625,
       38.5,
       22.28125,
       -15.21875,
       25.296875,
       -0.1875,
       28.671875,
       16.609375,
       20.1875,
       -11.375,
       8.09375,
       -13.453125,
       25.921875,
       11.703125,
       25.390625,
       8.734375,
       -33.34375,
       -6.9375,
       -37.859375,
       24.421875,
       3.859375,
       36.8125,
       11.671875,
       7.21875,
       -18.84375,
       -8.328125,
       0.609375,
       -0.40625,
       30.984375,
       -14.65625,
       12.75,
       -16.4375,
       -8.65625,
       -12.625,
       14.640625,
       47.9375,
       -14.5,
       34.84375,
       -23.34375,
       6.8125,
       -6.90625,
       -1.390625,
       22.34375,
       -3.859375,
       48.71875,
       -13.958333015441895,
       17.58333396911621,
       -2.6458332538604736,
       -12.854166984558105,
       7.041666507720947,
       -15.291666984558105,
       34.66666793823242,
       45.875,
       10.75,
       -17.02083396911621,
       -6.145833492279053,
       -32.60416793823242,
       23.64583396911621,
       3.7916667461395264,
       39.6875,
       17.375,
       -21.71875,
       -23.5625,
       -12.515625,
       3.5,
       9.71875,
       14.65625,
       -8.015625,
       -32.171875,
       -18.03125,
       -12.96875,
       -9.359375,
       14.25,
       39.8125,
       -17.78125,
       -31.828125,
       -24.34375,
       5.15625,
       -1.484375,
       19.375,
       25.09375,
       -2.953125,
       -25.875,
       -31.921875,
       27.546875,
       -17.515625,
       10.625,
       6.21875,
       -30.5,
       -23.875,
       -10.03125,
       28.890625,
       -2.796875,
       -18.03125,
       16.90625,
       -14.9375,
       -22.703125,
       -21.984375,
       53.171875,
       6.484375,
       15.3125,
       2.15625,
       -14.1875,
       -3.84375,
       -6.328125,
       25.46875,
       5.75,
       23.5625,
       -12.59375,
       -30.8125,
       8.28125,
       17.75,
       36.265625,
       19.796875,
       36.25,
       17.0625,
       -24.625,
       16.0625,
       17.6875,
       15.25,
       11.453125,
       82.59375,
       7.71875,
       -32.3125,
       26.390625,
       14.59375,
       14.484375,
       -5.296875,
       65.28125,
       -12.9375,
       -22.15625,
       35.859375,
       11.59375,
       34.40625,
       -8.90625,
       76.46875,
       -3.7916667461395264,
       -24.5,
       31.39583396911621,
       -2.2291667461395264,
       21.70833396911621,
       -11.520833015441895,
       70.75,
       -68.125,
       22.0625,
       -17.95833396911621,
       -11.270833015441895,
       -36.33333206176758,
       -46.97916793823242,
       -34.54166793823242,
       -38.6875,
       27.0,
       -14.03125,
       2.78125,
       -35.6875,
       -43.265625,
       -29.8125,
       -6.59375,
       24.78125,
       -25.75,
       -14.34375,
       -42.78125,
       -36.171875,
       -7.65625,
       22.0625,
       31.21875,
       -17.609375,
       7.109375,
       -1.734375,
       -14.96875,
       -0.90625,
       53.75,
       37.515625,
       -38.8125,
       7.265625,
       8.015625,
       -2.09375,
       2.15625,
       46.25,
       23.671875,
       -33.75,
       11.59375,
       24.4375,
       20.125,
       23.96875,
       11.75,
       13.859375,
       -36.84375,
       8.953125,
       63.65625,
       33.984375,
       22.6875,
       -17.375,
       21.375,
       -29.328125,
       20.1875,
       33.609375,
       6.921875,
       59.9375,
       -44.1875,
       27.75,
       -5.640625,
       43.1875,
       0.203125,
       14.578125,
       47.125,
       -65.625,
       31.71875,
       -30.75,
       28.71875,
       -7.984375,
       -6.171875,
       23.9375,
       -42.71875,
       48.078125,
       -23.453125,
       31.59375,
       -24.859375,
       -4.59375,
       19.96875,
       -17.75,
       10.4375,
       -26.140625,
       6.046875,
       -30.9375,
       17.421875,
       -10.84375,
       -9.333333015441895,
       2.0625,
       -39.02083206176758,
       -23.70833396911621,
       -19.27083396911621,
       19.5,
       -16.5});
  op_avg_pool2d_out(
      self,
      kernel_size,
      stride,
      padding,
      ceil_mode,
      count_include_pad,
      divisor_override,
      out);
  EXPECT_TENSOR_CLOSE(out, out_expected);
}