Function create_test_node

internal/bin/test_permute_op.cpp:88–164

}

int create_test_node(graph_t graph, const char* node_name, int order, const char* inputs[], int input_counts)
{
    std::cout << "create_ggraph_node--" << order << "\n";
    node_t test_node = create_graph_node(graph, node_name, "Permute");
    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};
    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);

        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);
    }

    int order0 = 0;
    int order1 = 2;
    int order2 = 3;
    int order3 = 1;
    std::cout << "layout : " << order << "\n";
    if(order == TENGINE_LAYOUT_NHWC)
    {
        order1 = 3;
        order2 = 1;
        order3 = 2;

        int tmp = dim[1];
        dim[1] = dim[3];
        dim[3] = dim[2];
        dim[2] = tmp;
    }
    else
    {
        int tmp = dim[1];
        dim[1] = dim[2];
        dim[2] = dim[3];
        dim[3] = tmp;
    }
    std::cout << "output_dim: " << dim[0] << "-" << dim[1] << "-" << dim[2] << "-" << dim[3] << "\n";
    std::cout << "orders:" << order0 << "-" << order1 << "-" << order2 << "-" << order3 << "\n";

    /* output */
    tensor_t output_tensor = create_graph_tensor(graph, node_name, data_type);