Function create_test_node

internal/bin/test_concat_op.cpp:88–138

}

int create_test_node(graph_t graph, const char* node_name, int axis, const char* inputs[], int input_counts)
{
    std::cout << "create_ggraph_node--" << axis << "\n";
    node_t test_node = create_graph_node(graph, node_name, "Concat");
    if(test_node == nullptr)
    {
        std::cout << "create_graph_node_failed : node_name=" << node_name << "\n";
        return -1;
    }

    int data_type = 0;
    int dim[4] = {0};
    int concat_dim = 0;
    for(int ii = 0; ii < input_counts; ++ii)
    {
        tensor_t input_tensor = get_graph_tensor(graph, inputs[ii]);
        data_type = get_tensor_data_type(input_tensor);

        get_tensor_shape(input_tensor, dim, 4);
        concat_dim += dim[axis];

        if(input_tensor == nullptr)
        {
            std::cout << "ERRNO: " << get_tengine_errno() << "\noutput";
            return -1;
        }
        // std::cout << "set_node_input_tensor\n";
        set_node_input_tensor(test_node, ii, input_tensor);
        // std::cout << "release_input_tensor\n";
        release_graph_tensor(input_tensor);
    }

    dim[axis] = concat_dim;
    std::cout << "output_dim: " << dim[0] << "-" << dim[1] << "-" << dim[2] << "-" << dim[3] << "\n";
    /* output */
    tensor_t output_tensor = create_graph_tensor(graph, node_name, data_type);
    set_tensor_shape(output_tensor, dim, 4);
    set_node_output_tensor(test_node, 0, output_tensor, TENSOR_TYPE_VAR);

    if(data_type == TENGINE_DT_UINT8)
    {
        set_tensor_quant_param(output_tensor, &uint8_scale, &uint8_zero, 1);
    }

    release_graph_tensor(output_tensor);

    set_node_attr_int(test_node, "axis", &axis);
    release_graph_node(test_node);

    return 0;
}

graph_t create_test_graph(const char* test_node_name, int axis, const std::vector<ConcatInputParam>& inputParam,
                          int layout, int data_type)