Method ForwardBFS

core/lib/graph.cpp:778–842

}

void Graph::ForwardBFS(Graph* graph, std::vector<Node*>& starts, graph_visit_t func, bool input_ready)
{
    int node_number = graph->seq_nodes.size();
    std::vector<int> visited(node_number, 0);
    std::set<Node*> in_graph;

    for(int i = 0; i < node_number; i++)
        in_graph.insert(graph->seq_nodes[i]);

    std::queue<Node*> visit_queue;

    /* inital the visit list */
    for(unsigned int i = 0; i < starts.size(); i++)
    {
        Node* node = starts[i];

        bool pure_input_node = true;

        // only if all input tensors are out-side graph

        for(int i = 0; i < node->GetParentNum(); i++)
        {
            Node* pnode = node->GetParentNode(i);

            if(in_graph.count(pnode))
            {
                pure_input_node = false;
                break;
            }
        }

        if(pure_input_node)
        {
            visit_queue.push(node);
            visited[node->GetNodeIndex()] = 1;
            func(graph, node);
        }
    }

    while(visit_queue.size())
    {
        Node* node = visit_queue.front();
        visit_queue.pop();

        int output_num = node->GetOutputNum();

        for(int i = 0; i < output_num; i++)
        {
            Tensor* tensor = node->GetOutputTensorSeq(i);

            for(unsigned int k = 0; k < tensor->consumer.size(); k++)
            {
                Node* child = tensor->consumer[k]->owner;

                if(in_graph.count(child) && !visited[child->GetNodeIndex()] &&
                   (!input_ready || AllInputVisited(graph, child, visited, in_graph)))
                {
                    visit_queue.push(child);