Function create_input_node

internal/bin/test_concat_op.cpp:57–86

const char* gInputName[] = {"data1", "data2", "data3", "data4", "data5"};

int create_input_node(graph_t graph, const char* node_name, int n, int c, int h, int w, int layout, int data_type)
{
    node_t node = create_graph_node(graph, node_name, "InputOp");
    tensor_t tensor = create_graph_tensor(graph, node_name, data_type);

    // printf("create_input_node [%d]:[%d]:[%d]:[%d]\n",n,c,h,w);
    if(tensor == nullptr)
    {
        release_graph_node(node);
        return -1;
    }

    set_node_output_tensor(node, 0, tensor, TENSOR_TYPE_INPUT);

    if(layout == 0)
    {
        int dims[4] = {n, c, h, w};
        set_tensor_shape(tensor, dims, 4);
    }
    else
    {
        int dims[4] = {n, h, w, c};
        set_tensor_shape(tensor, dims, 4);
    }

    release_graph_tensor(tensor);
    release_graph_node(node);

    return 0;
}

int create_test_node(graph_t graph, const char* node_name, int axis, const char* inputs[], int input_counts)
{