Function parse_node

driver/cpu/cpu_runner.cpp:266–338

#ifdef ENABLE_TIME_PROFILING

static void parse_node(void* data, int repeat_count, uint64_t total_time)
{
    const char* env = std::getenv("DEBUG");
    Node* node = ( Node* )data;
    NodeOps* node_ops = any_cast<NodeOps*>(node->GetAttr(ATTR_NODE_OPS));
    Tensor* input_tensor = node->GetInputTensor(0);
    Tensor* output_tensor = node->GetOutputTensor(0);
    Operator* op = node->GetOp();
    TShape& ishape = input_tensor->GetShape();
    const std::vector<int>& in_dim = ishape.GetDim();
    TShape& oshape = output_tensor->GetShape();
    const std::vector<int>& out_dim = oshape.GetDim();
    printf(" Node_idx: %4d", node->GetNodeIndex());
    // float* outdata=(float *)get_tensor_mem(output_tensor);
    // float* indata=(float *)get_tensor_mem(input_tensor);
    // printf("\t%s\n",output_tensor->GetName().c_str());

    printf("  %10s", op->GetName().c_str());
    if(env && env[0] == '1')
        printf("(%s)", node_ops->GetName().c_str());
    printf("\t\t");
    for(size_t i = 0; i < in_dim.size() - 1; i++)
    {
        printf("%dx", in_dim[i]);
    }
    printf("%d -> ", in_dim[in_dim.size() - 1]);
    for(size_t i = 0; i < out_dim.size() - 1; i++)
    {
        printf("%dx", out_dim[i]);
    }
    printf("%d\t", out_dim[in_dim.size() - 1]);
    if(op->GetName() == "Convolution" || op->GetName() == "Deconvolution")
    {
        Convolution* conv_op = dynamic_cast<Convolution*>(node->GetOp());
        ConvParam* param = conv_op->GetParam();
        printf("K: %dx%d | S: %dx%d | P: %d %d %d %d", param->kernel_h, param->kernel_w, param->stride_h, param->stride_w,
               param->pad_h0, param->pad_h1, param->pad_w0, param->pad_w1);
        if(param->group != 1)
        {
            printf(" DW(%2d)", param->group);
        }
        else
        {
            printf("       ");
        }
    }
    else if(op->GetName() == "Pooling")
    {
        Pooling* conv_op = dynamic_cast<Pooling*>(node->GetOp());
        PoolParam* param = conv_op->GetParam();
        printf("K: %dx%d | S: %dx%d | P: %d %d %d %d", param->kernel_h, param->kernel_w, param->stride_h, param->stride_w,
               param->pad_h0, param->pad_h1, param->pad_w0, param->pad_w1);
        if(param->alg == 0)
        {
            printf("       Max");
        }
        else
        {