Function create_pool_graph

internal/bin/test_pooling_int8.cpp:80–124

}

graph_t create_pool_graph(int c, int h, int w, int k_h, int k_w, int s_h, int s_w, int pad_h0, int pad_w0, int pad_h1,
                          int pad_w1, int m)
{
    graph_t graph = create_graph(nullptr, nullptr, nullptr);

    if(graph == nullptr)
    {
        std::cerr << "ERRNO: " << get_tengine_errno() << "\n";
        return nullptr;
    }

    const char* input_name = "data";
    const char* pool_name = "pool";

    if(create_input_node(graph, input_name, c, h, w) < 0)
    {
        std::cerr << "create input failed\n";
        return nullptr;
    }

    if(create_pool_node(graph, pool_name, input_name, k_h, k_w, s_h, s_w, pad_h0, pad_w0, pad_h1, pad_w1, m) < 0)
    {
        std::cerr << "create pool node failed\n";
        return nullptr;
    }

    /* set input/output node */

    const char* inputs[] = {input_name};
    const char* outputs[] = {pool_name};

    if(set_graph_input_node(graph, inputs, sizeof(inputs) / sizeof(char*)) < 0)
    {
        std::cerr << "set inputs failed: ERRNO: " << get_tengine_errno() << "\n";
        return nullptr;
    }

    if(set_graph_output_node(graph, outputs, sizeof(outputs) / sizeof(char*)) < 0)
    {
        std::cerr << "set outputs failed: ERRNO: " << get_tengine_errno() << "\n";
        return nullptr;
    }

    return graph;
}

int test_pool(int c, int h, int w, int k_h, int k_w, int s_h, int s_w, int pad_h0, int pad_w0, int pad_h1, int pad_w1,
              int m)