Function create_test_node

internal/bin/test_split.cpp:58–103

    return 0;
}
int create_test_node(graph_t graph, const char* node_name, int axis, int numbe_split, int dim0, int dim1, int dim2,
                     const char* input_name)
{
    node_t test_node = create_graph_node(graph, node_name, "Split");

    const int c_axis = axis;
    set_node_attr_int(test_node, "axis", &c_axis);
    if(numbe_split == 0)
    {
        const std::vector<int> slice_dims = {dim0, dim1, dim2};
        set_node_attr_pointer(test_node, "split_sizes_", &slice_dims);
    }
    else
    {
        const int slice_num = numbe_split;
        set_node_attr_int(test_node, "split_dim", &slice_num);
    }

    tensor_t input_tensor = get_graph_tensor(graph, input_name);
    int data_type = get_tensor_data_type(input_tensor);

    if(input_tensor == nullptr)
    {
        std::cout << "ERRNO: " << get_tengine_errno() << "\n";
        return -1;
    }

    set_node_input_tensor(test_node, 0, input_tensor);

    release_graph_tensor(input_tensor);

    /* output */

    tensor_t output_tensor_0 = create_graph_tensor(graph, "t0", data_type);
    tensor_t output_tensor_1 = create_graph_tensor(graph, "t1", data_type);
    tensor_t output_tensor_2 = create_graph_tensor(graph, "t2", data_type);
    set_node_output_tensor(test_node, 0, output_tensor_0, TENSOR_TYPE_VAR);
    set_node_output_tensor(test_node, 1, output_tensor_1, TENSOR_TYPE_VAR);
    set_node_output_tensor(test_node, 2, output_tensor_2, TENSOR_TYPE_VAR);
    release_graph_tensor(output_tensor_0);
    release_graph_tensor(output_tensor_1);
    release_graph_tensor(output_tensor_2);
    release_graph_node(test_node);

    return 0;
}

graph_t create_test_graph(const char* test_node_name, int axis, int numbe_split, int dim0, int dim1, int dim2, int c,
                          int h, int w, int layout, int data_type)