ncnn/tests/test_fold.cpp

// Copyright 2022 Tencent
// SPDX-License-Identifier: BSD-3-Clause

#include "testutil.h"

#include "layer_type.h"

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)
{
    ncnn::Mat a = RandomMat(w, h);

    ncnn::ParamDict pd;
    pd.set(1, kernel_w);
    pd.set(11, kernel_h);
    pd.set(2, dilation_w);
    pd.set(12, dilation_h);
    pd.set(3, stride_w);
    pd.set(13, stride_h);
    pd.set(4, pad_w);
    pd.set(14, pad_h);
    pd.set(20, outw);
    pd.set(21, outh);

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

    int ret = test_layer("Fold", pd, weights, a);
    if (ret != 0)
    {
        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);
    }

    return ret;
}

static int test_fold_0()
{
    return 0
           || test_fold(400, 108, 22, 22, 3, 3, 1, 1, 1, 1, 0, 0)
           || test_fold(190, 96, 18, 17, 4, 2, 1, 1, 1, 2, 2, 2)
           || test_fold(120, 36, 11, 5, 3, 2, 2, 1, 1, 1, 4, 2);
}

#if NCNN_VALIDATION
static int test_fold_load_param()
{
    ncnn::ParamDict base;
    base.set(1, 3);
    if (test_layer_param(ncnn::LayerType::Fold, base, 0) != 0)
        return -1;

    if (test_layer_param(ncnn::LayerType::Fold, base, 3, 0, -1) != 0)
        return -1;

    {
        ncnn::ParamDict pd = base;
        pd.set(3, 1);
        pd.set(1, 65536);
        if (test_layer_param(ncnn::LayerType::Fold, pd, -1) != 0)
            return -1;
    }

    return 0;
}
#endif // NCNN_VALIDATION

int main()
{
    SRAND(7767517);

    return 0
           || test_fold_0()
#if NCNN_VALIDATION
           || test_fold_load_param()
#endif // NCNN_VALIDATION
           ;
}