Method Run

executor/operator/ref/ref_logical.cpp:51–113

}

bool LogicalOps::Run(Node* node)
{
    Logical* logical_op = dynamic_cast<Logical*>(node->GetOp());
    LogicalParam* param = logical_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;

    // int x = node->GetInputNum();
    // printf("inputnum:%d\n", x);
    // 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);
        // printf("input_tensor1 %p\n", input_tensor1);
        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;

    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);
    // }