Function main

internal/bin/test_permute_op.cpp:533–575

}

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", "Layout", "Res"};
    CVSResOut ResOut("permute_ref_test.cvs", pTitle, 3);

    test(layout, data_type, ResOut);

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

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

    // uint8
    data_type++;
    test(layout, data_type, ResOut);

    ResOut.save();
    release_tengine();

    return 0;
}