Function main

internal/bin/test_deconv.cpp:144–384

}

int main(int argc, char* argv[])
{
    int c, h, w, k, s, p, o_c, g;

    int rep = 1;
    c = 32;
    w = 20;
    k = 4;
    s = 2;
    p = 1;
    o_c = 64;
    g = 1;


    int res;
    while((res = getopt(argc, argv, "c:w:k:s:r:h")) != -1)
    {
        switch(res)
        {
            case 'c':
                c = atoi(optarg);
                break;
            case 'w':
                w = atoi(optarg);
                break;
            case 'k':
                k = atoi(optarg);
                break;
            case 's':
                s = atoi(optarg);
                break;
            case 'r':
                rep = atoi(optarg);
                break;
            case 'p':
                p = atoi(optarg);
                break;
            case 'o':
                o_c = atoi(optarg);
                break;
            case 'g':
                g = atoi(optarg);
                break;
            case 'h':
                std::cout   << "[Usage]: " << argv[0] << " [-h]\n"
                            << "    [-c input_c] [-w input_w] [-k kernel] [-s stride] [-r repeat]\n";
                return 0;
            default:
                break;
        }
    }
    h = w;

    init_tengine();

    graph_t graph = create_conv_graph(c, h, w, o_c, k, s, p, g);
    graph_t graph1 = create_conv_graph(c, h, w, o_c, k, s, p, g);