Function Weight_Bn

executor/lib/graph_optimizer.cpp:47–199

static void AddConstNodeToSubGraph(Subgraph* graph, Tensor* tensor, Node* fused_node, int fused_port_index);

static bool Weight_Bn(Subgraph* graph, Node* ConvNode, float* mean, float* var, float* gamma, float* beta, float eps,
                      float rescale_factor, Tensor* bias_tensor)
{
    Tensor* kernel_tensor = ConvNode->GetInputTensor(1);
    Convolution* conv_op = dynamic_cast<Convolution*>(ConvNode->GetOp());
    ConvParam* param = conv_op->GetParam();
    const TShape& kernel_shape = kernel_tensor->GetShape();

    int group = param->group;
    // int input_chan = kernel_shape.Shape(1);
    int input_chan = kernel_shape.GetC();
    int output_chan = kernel_shape.Shape(0) / group;
    int kernel_x = param->kernel_w;
    int kernel_y = param->kernel_h;
    int kernel_size = input_chan * kernel_x * kernel_y;
    float* kernel_org = ( float* )get_tensor_mem(kernel_tensor);
    int channel_num = kernel_shape.Shape(0);
    float* kernel_new = ( float* )(malloc(kernel_size * channel_num * sizeof(float) + 128));

    memcpy(kernel_new, kernel_org, sizeof(float) * kernel_size * channel_num);

    kernel_tensor->SetMemAddr(kernel_new);
    kernel_tensor->SetAttr("free_mem", 1);

    float* scale_mean = ( float* )malloc(channel_num * sizeof(float));
    float* scale_var_inv = ( float* )malloc(channel_num * sizeof(float));

    float rescale_factor_tmp = rescale_factor;

    // fuse the bias;
    float* bias = NULL;
    std::string bias_name;
    if(bias_tensor)
    {
        bias = ( float* )get_tensor_mem(bias_tensor);
        bias_name = bias_tensor->GetName() + ".bn";
    }
    else
    {
        bias_name = ConvNode->GetName() + ".bias.bn";
    }

    /*
     *  create the bias node,,, ugly code..
     *
     */

    {
        Tensor* new_bias_tensor = new Tensor(bias_name);
        std::vector<int> dims{channel_num};

        TShape bias_shape;
        bias_shape.SetDim(dims);

        new_bias_tensor->Reshape(bias_shape);
        new_bias_tensor->SetType(kConstTensor);

        void* bias_new = ( void* )malloc(channel_num * sizeof(float) + 128);