Function GraphFuseConvBN

executor/lib/graph_optimizer.cpp:449–583

}

static bool GraphFuseConvBN(Graph* graph, GraphOptimizer* opt)
{
    int node_number = graph->seq_nodes.size();
    std::vector<Subgraph*> orig_sub;

    /*get all bn_scale chain*/
    for(int i = 0; i < node_number; i++)
    {
        Node* Bn_node = graph->seq_nodes[i];
        Operator* op = Bn_node->GetOp();

        if(op->GetName() != "BatchNormalization")
            continue;

        /*Check if it is a Conv-->Bn*/
        Tensor* input_tensor;
        Node* Conv_node;

        input_tensor = Bn_node->GetInputTensor(0);
        Conv_node = input_tensor->producer->owner;
        op = Conv_node->GetOp();

        if(op->GetName() != "Convolution")
            continue;

        /*Create a subgrah represent the chain*/
        Subgraph* sub = new Subgraph("ConvBn_chain");

        sub->seq_nodes.push_back(Conv_node);
        sub->seq_nodes.push_back(Bn_node);

        sub->input_nodes.push_back(Conv_node);
        sub->output_nodes.push_back(Bn_node);

        /* add const node into seq nodes */
        for(unsigned int i = 1; i < Conv_node->GetInputNum(); i++)
        {
            Tensor* tensor = Conv_node->GetInputTensor(i);
            sub->seq_nodes.push_back(tensor->producer->owner);
        }

        for(unsigned int i = 1; i < Bn_node->GetInputNum(); i++)
        {
            Tensor* tensor = Bn_node->GetInputTensor(i);
            sub->seq_nodes.push_back(tensor->producer->owner);
        }
        orig_sub.push_back(sub);
    }

    /*replace the nodes of the graph*/
    for(unsigned int i = 0; i < orig_sub.size(); i++)
    {
        Subgraph fused("fused");
        Subgraph* orig = orig_sub[i];

        Node* orig_output = orig->output_nodes[0];
        Node* orig_input = orig->input_nodes[0];