Function create_fc_node

internal/tests/executor/tests/test_fc.cpp:53–174

}

Node* create_fc_node(void)
{
    Operator* op = OpManager::CreateOp("FullyConnected");
    FullyConnected* fc_op = dynamic_cast<FullyConnected*>(op);

    FCParam* param = fc_op->GetParam();

    /* calculate shapes */

#if 1
    param->num_output = 2;
    int input_n = 1;
    int input_c = 128;
#else
    param->num_output = 12544;
    int input_n = 32;
    int input_c = 288;
#endif

    int input_h = 1;
    int input_w = 1;

    int N = input_n;
    int K = input_h * input_w * input_c;
    int M = param->num_output;

    std::vector<int> input_dims = {input_n, input_c, input_h, input_w};
    std::vector<int> weight_dims = {M, K};
    std::vector<int> bias_dims = {M};
    std::vector<int> output_dims = {N, M};

    op_fops = 1.0 * N * M * (2 * K);

    std::cout << "Input n: " << input_n << " c: " << input_c << " h: " << input_h << " w: " << input_w << "\n";
    std::cout << "weight M: " << M << " K: " << K << "\n";
    std::cout << "N: " << N << " M: " << M << " K: " << K << "\n";

    Node* node = new Node("test_fc");

    node->SetOp(fc_op);

    // prepare tensor: input/weight/bias/output
    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);