Function create_scale_node

internal/tests/executor/tests/test_scale.cpp:71–156

float op_fops;

Node* create_scale_node(int n, int c, int h, int w)
{
    Operator* op = OpManager::CreateOp("Scale");
    Scale* scale_op = dynamic_cast<Scale*>(op);

    ScaleParam* param = scale_op->GetParam();
    param->axis = 1;
    param->bias_term = 1;
    param->num_axes = 1;
    /* 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(scale_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("gamma");
    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);
    /*** beta ***/

    tensor = new Tensor("beta");
    tensor->SetDataType("float32");
    tensor->SetType(kVarTensor);
    shape = &tensor->GetShape();
    shape->SetDataLayout("W");