Function get_driving_nodes

src/graph/Utils.cpp:180–198

}

std::vector<NodeIdxPair> get_driving_nodes(const INode &node)
{
    std::vector<NodeIdxPair> driving_nodes;

    const Graph *g = node.graph();
    ARM_COMPUTE_ERROR_ON(g == nullptr);

    for (auto &output_edge_id : node.output_edges())
    {
        auto output_edge = g->edge(output_edge_id);
        if (output_edge != nullptr)
        {
            ARM_COMPUTE_ERROR_ON(output_edge->consumer() == nullptr);
            driving_nodes.push_back({output_edge->consumer_id(), output_edge->consumer_idx()});
        }
    }

    return driving_nodes;
}

std::vector<NodeIdxPair> get_driver_nodes(const INode &node)
{