Function test

internal/bin/test_split.cpp:289–353

    }
}
int test(const char* test_node_name, int c, int h, int w, int layout, int axis, int number_split, int dim0, int dim1,
         int dim2, int data_type)
{
    std::cout << "\n\n-----------------------------------------axis: " << axis << "  number_split: " << number_split
              << "  data_type: " << data_type << " splitdims: " << dim0 << "," << dim1 << "," << dim2
              << "----------------------------------\n\n";
    graph_t graph = create_test_graph(test_node_name, axis, number_split, dim0, dim1, dim2, c, h, w, layout, data_type);

    if(graph == nullptr)
        return 1;

    /* set input */
    void* i_buf = set_input_data(graph);
    // void * s_buf=set_input_slope_data(graph);
    // if(data_type == 3)
    // {
    //     tensor_t output = get_graph_output_tensor(graph, 0, 0);
    //     float scale = 0.1;
    //     int zero = 30;
    //     set_tensor_quant_param(output,&scale,&zero,1);
    // }
    // if(data_type == 2)
    // {
    //     tensor_t output = get_graph_output_tensor(graph, 0, 0);
    //     float scale = 0.1;
    //     int zero = 0;
    //     set_tensor_quant_param(output,&scale,&zero,1);
    // }
    tensor_t output = get_graph_output_tensor(graph, 0, 0);
    if(data_type == 3)
    {
        float scale = 0.1;
        int zero = 0;
        set_tensor_quant_param(output, &scale, &zero, 1);
    }
    // prerun graph
    if(prerun_graph(graph) < 0)
    {
        std::cerr << "prerun_graph failed: ERRNO: " << get_tengine_errno() << "\n";
        return 1;
    }
    dump_graph(graph);
    node_t test_node = get_graph_node(graph, test_node_name);

    const char* dev = get_node_device(test_node);

    std::cout << "node running on dev: " << dev << "\n";

    if(run_graph(graph, 1) < 0)
    {
        std::cerr << "run_graph failed: ERRNO: " << get_tengine_errno() << "\n";
        return 2;
    }

    dump_output_data(test_node);

    free(i_buf);