Function LoadPool

tengine-module/plugin/serializer/tensorflow/tf_serializer.cpp:2426–2482

    return true;
}
static bool LoadPool(TFNode* tf_node, TFGraph& tf_graph, StaticGraph* graph)
{
    TFNode* input = tf_node->inputs[0];
    StaticNode* node = tf_node->static_node;

    AddNodeInputTensor(node, input->static_tensor);

    PoolParam param = any_cast<PoolParam>(OpManager::GetOpDefParam("Pooling"));

    const tensorflow::NodeDef* node_def = tf_node->pb_defs[0];
    tensorflow::AttrValue value;

    if(GetAttrValue(node_def, "ksize", value))
    {
        param.kernel_h = value.list().i(1);
        param.kernel_w = value.list().i(2);
    }

    if(GetAttrValue(node_def, "strides", value))
    {
        param.stride_h = value.list().i(1);
        param.stride_w = value.list().i(2);
    }

    if(GetAttrValue(node_def, "padding", value))
    {
        if(value.s() == "VALID")
        {
            param.pad_h0 = 0;
            param.pad_h1 = 0;
            param.pad_w0 = 0;
            param.pad_w1 = 0;
        }
        else if(value.s() == "SAME")
        {
            param.pad_h0 = -1;
            param.pad_h1 = -1;
            param.pad_w0 = -1;
            param.pad_w1 = -1;
        }
    }

    if(tf_node->op == "AvgPool")
    {
        param.alg = kPoolAvg;
    }
    else if(tf_node->op == "MaxPool")
    {
        param.alg = kPoolMax;
    }

    StaticOp* op = CreateStaticOp(graph, "Pooling");
    SetOperatorParam(op, param);
    SetNodeOp(node, op);

    return true;
}