Function create_prelu_node

internal/tests/executor/tests/test_prelu.cpp:66–125

float op_fops;

Node* create_prelu_node(int n, int c, int h, int w)
{
    Operator* op = OpManager::CreateOp("PReLU");
    PReLU* prelu_op = dynamic_cast<PReLU*>(op);

    /* input shape */
    int input_h = h;
    int input_w = w;
    int input_c = c;
    int input_n = n;

    std::vector<int> input_dims = {input_n, input_c, input_h, input_w};
    std::vector<int> channel_dims = {input_c};

    Node* node = new Node("test Scale");

    node->SetOp(prelu_op);

    // prepare tensor: input/gmma/beta
    Tensor* tensor;
    int mem_size;
    void* addr;

    tensor = new Tensor("input");

    tensor->SetDataType("float32");
    tensor->SetType(kVarTensor);
    TShape* shape = &tensor->GetShape();
    shape->SetDataLayout("NCHW");
    shape->SetDim(input_dims);
    node->SetInputPort(0, tensor);
    mem_size = tensor->GetTotalSize();
    addr = std::malloc(mem_size);
    init_tensor_data(( float* )addr, mem_size / sizeof(float), -1);
    set_tensor_mem(tensor, addr, mem_size, std::free);

    tensor = new Tensor("slope");
    tensor->SetDataType("float32");
    tensor->SetType(kVarTensor);
    shape = &tensor->GetShape();
    shape->SetDataLayout("W");
    shape->SetDim(channel_dims);    //
    node->SetInputPort(1, tensor);
    mem_size = tensor->GetTotalSize();
    addr = std::malloc(mem_size);
    init_tensor_data(( float* )addr, mem_size / sizeof(float), -3);
    set_tensor_mem(tensor, addr, mem_size, std::free);

    tensor = new Tensor("output");
    tensor->SetDataType("float32");
    tensor->SetType(kVarTensor);
    shape = &tensor->GetShape();
    shape->SetDataLayout("NCHW");
    shape->SetDim(input_dims);
    node->SetOutputPort(0, tensor);
    mem_size = tensor->GetTotalSize();
    addr = std::malloc(mem_size);
    set_tensor_mem(tensor, addr, mem_size, std::free);