Method RealCreateFromStatic

core/lib/graph.cpp:279–351

#endif

bool Graph::RealCreateFromStatic(const StaticGraphPtr& static_graph)
{
    orig_graph_ = static_graph;
    attrs_ = static_graph->attrs;

    int node_number = static_graph->node_list.size();

    /* create node and its output tensor */
    for(int i = 0; i < node_number; i++)
    {
        const StaticNode* node_ptr = static_graph->node_list[i].get();

        Node* node = new Node(node_ptr->name);

        if(!CreateNodeFromStatic(node, static_graph.get(), node_ptr))
            return false;
    }

    /* Setup the connections */
    int tensor_number = static_graph->tensor_list.size();

    for(int i = 0; i < tensor_number; i++)
    {
        const StaticTensor* static_tensor = static_graph->tensor_list[i].get();
        Tensor* tensor = FindTensor(static_tensor->name);

        if(tensor == nullptr)
        {
            XLOG_ERROR() << "cannot find tensor: " << static_tensor->name << "\n";
            return false;
        }

        if(!SetupConnection(tensor, static_graph.get(), static_tensor))
            return false;
    }

    /* set the input and output */

    for(unsigned int i = 0; i < static_graph->input_node_list.size(); i++)
    {
        int node_idx = static_graph->input_node_list[i];
        Node* node = FindNode(static_graph->node_list[node_idx].get()->name);
        input_nodes.push_back(node);

        /* update the input node's tensor type */
        Tensor* tensor = node->GetOutputTensor(0);
        tensor->SetType(kInputTensor);
    }

    for(unsigned int i = 0; i < static_graph->output_node_list.size(); i++)
    {
        int node_idx = static_graph->output_node_list[i];
        Node* node = FindNode(static_graph->node_list[node_idx].get()->name);
        output_nodes.push_back(node);
    }

    /* re-order the seq_nodes_, removing un-used node, according to node dependency*/