/**
 * \file dnn/test/naive/sliding_window_transpose.cpp
 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
 *
 * Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 */
#include "test/naive/fixture.h"

#include "megdnn/oprs/nn.h"
#include "test/common/checker.h"

using namespace megdnn;
using namespace test;

TEST_F(NAIVE, SlidingWindowTranspose_FORWARD) {
    Checker<SlidingWindowTranspose> checker(handle(), /* check_dispatch */false);
    
    SlidingWindowTranspose::Param param(3,3,0,0,1,1,1,1,2,2);
    checker.set_param(param).exect(
            Testcase{TensorValue({1, 1, 2, 2, 2, 2}, dtype::Uint8(),
                               {0,1,3,4,
                                1,2,4,5,
                                3,4,6,7,
                                4,5,7,8}), {}}, 
            Testcase{{},
            TensorValue({1, 1, 3, 3}, dtype::Uint8(), 
                        {0,2,2,
                        6,16,10,
                        6,14,8})});
    
    param.out_h = 6;
    param.out_w = 7;
    param.pad_h = 1;
    param.pad_w = 1;
    param.stride_h = 2;
    param.stride_w = 2;
    param.dilate_h = 2;
    param.dilate_w = 2;
    param.window_h = 3;
    param.window_w = 3;
    checker.set_param(param).exect(
          Testcase{TensorValue({1, 1, 2, 3, 3, 3}, dtype::Uint8(),
                               {0,0,0,0,8,10,0,22,24,
                                0,0,0,8,10,12,22,24,26,
                                0,0,0,10,12,0,24,26,0,
                                0,8,10,0,22,24,0,36,38,
                                8,10,12,22,24,26,36,38,40,
                                10,12,0,24,26,0,38,40,0}), {}}, 
          Testcase{{},
          TensorValue({1, 1, 6, 7}, dtype::Uint8(),
                        {0,0,0,0,0,0,0,
                         0,32,0,60,0,48,0,
                         0,0,0,0,0,0,0,
                         0,88,0,144,0,104,0,
                         0,0,0,0,0,0,0,
                         0,72,0,114,0,80,0})});
}