ncnn

1
// Tencent is pleased to support the open source community by making ncnn available.
2
//
3
// Copyright (C) 2022 THL A29 Limited, a Tencent company. All rights reserved.
4
//
5
// Licensed under the BSD 3-Clause License (the "License"); you may not use this file except
6
// in compliance with the License. You may obtain a copy of the License at
7
//
8
// https://opensource.org/licenses/BSD-3-Clause
9
//
10
// Unless required by applicable law or agreed to in writing, software distributed
11
// under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
12
// CONDITIONS OF ANY KIND, either express or implied. See the License for the
13
// specific language governing permissions and limitations under the License.
14

15
#include "testutil.h"
16

17
static int test_fold(int w, int h, int outw, int outh, int kernel_w, int kernel_h, int dilation_w, int dilation_h, int stride_w, int stride_h, int pad_w, int pad_h)
18
{
19
    ncnn::Mat a = RandomMat(w, h);
20

21
    ncnn::ParamDict pd;
22
    pd.set(1, kernel_w);
23
    pd.set(11, kernel_h);
24
    pd.set(2, dilation_w);
25
    pd.set(12, dilation_h);
26
    pd.set(3, stride_w);
27
    pd.set(13, stride_h);
28
    pd.set(4, pad_w);
29
    pd.set(14, pad_h);
30
    pd.set(20, outw);
31
    pd.set(21, outh);
32

33
    std::vector<ncnn::Mat> weights(0);
34

35
    int ret = test_layer("Fold", pd, weights, a);
36
    if (ret != 0)
37
    {
38
        fprintf(stderr, "test_fold failed w=%d h=%d outw=%d outh=%d kernel=%d,%d dilation=%d,%d stride=%d,%d pad=%d,%d\n", w, h, outw, outh, kernel_w, kernel_h, dilation_w, dilation_h, stride_w, stride_h, pad_w, pad_h);
39
    }
40

41
    return ret;
42
}
43

44
static int test_fold_0()
45
{
46
    return 0
47
           || test_fold(400, 108, 22, 22, 3, 3, 1, 1, 1, 1, 0, 0)
48
           || test_fold(190, 96, 18, 17, 4, 2, 1, 1, 1, 2, 2, 2)
49
           || test_fold(120, 36, 11, 5, 3, 2, 2, 1, 1, 1, 4, 2);
50
}
51

52
int main()
53
{
54
    SRAND(7767517);
55

56
    return test_fold_0();
57
}
58