Method Run

executor/operator/ref/ref_eltwise.cpp:128–188

}

bool EltwiseOps::Run(Node* node)
{
    Eltwise* elt_op = dynamic_cast<Eltwise*>(node->GetOp());
    EltwiseParam* param = elt_op->GetParam();
    Tensor* input_tensor0 = node->GetInputTensor(0);
    const TShape& ishape = input_tensor0->GetShape();
    void* input0 = get_tensor_mem(input_tensor0);
    Tensor* input_tensor1 = nullptr;
    void* input1 = nullptr;
    // this version only support for input_num=2
    // int input_number=node->GetInputNum();

    // output
    Tensor* output_tensor = node->GetOutputTensor(0);
    if(input_tensor0->GetDataType() == TENGINE_DT_INT8 || input_tensor0->GetDataType() == TENGINE_DT_UINT8)
    {
        if(get_scale_zero(input_tensor0, output_tensor, &op_param) < 0)
            return false;
    }
   
    if(node->GetInputNum() > 1)
    {
        input_tensor1 = node->GetInputTensor(1);
        const TShape& ishape1 = input_tensor1->GetShape();
        input1 = get_tensor_mem(input_tensor1);
        op_param.shape1[0] = ishape1.GetN();
        op_param.shape1[1] = ishape1.GetC();
        op_param.shape1[2] = ishape1.GetH();
        op_param.shape1[3] = ishape1.GetW();

        if(input_tensor1->GetDataType() == TENGINE_DT_INT8 || input_tensor1->GetDataType() == TENGINE_DT_UINT8)
        {
            if(get_scale_zero_1(input_tensor1, &op_param) < 0)
                return false;
        }
    }
    void* output = get_tensor_mem(output_tensor);
    op_param.shape0[0] = ishape.GetN();
    op_param.shape0[1] = ishape.GetC();
    op_param.shape0[2] = ishape.GetH();
    op_param.shape0[3] = ishape.GetW();
    op_param.type = param->type;
    op_param.shift = param->shift;
    op_param.power = param->power;
    op_param.pScale = param->scale;
    int ret = kernel_run(input0, input1, output, &op_param);

    if(input_tensor0->GetDataType() == TENGINE_DT_INT8)
    {
        auto* o_quant = output_tensor->GetQuantParam();
        QuantParam q_param;
        q_param.scale = op_param.scale[2];
        o_quant->resize(0);
        o_quant->push_back(q_param);
    }

    if(ret < 0)
        return false;