Function main

internal/bin/test_concat_op.cpp:529–581

}

int main(int argc, char* argv[])
{
    int data_type = TENGINE_DT_FP32;
    int layout = TENGINE_LAYOUT_NCHW;

    int res;
    while((res = getopt(argc, argv, "c:h:w:n:s:p:m:l:t:")) != -1)
    {
        switch(res)
        {
            case 't':
                data_type = strtoul(optarg, NULL, 10);
                break;
            default:
                break;
        }
    }

    std::cout << "init_tengine\n";
    init_tengine();

    const char* pTitle[] = {"DatType", "Axis", "ConcatCounts", "Res"};
    CVSResOut ResOut("concat_ref_test.cvs", pTitle, 4);

    std::vector<int> vDims;
    vDims.push_back(2);
    vDims.push_back(3);
    vDims.push_back(2);
    vDims.push_back(2);

    for(int axis = 0; axis < 4; ++axis)
    {
        data_type = 0;
        test(layout, axis, data_type, vDims, ResOut);

        // fp16
        data_type++;
        test(layout, axis, data_type, vDims, ResOut);

        // int8
        data_type++;
        test(layout, axis, data_type, vDims, ResOut);

        // uint8
        data_type++;
        test(layout, axis, data_type, vDims, ResOut);
    }

    ResOut.save();
    release_tengine();

    return 0;
}