Function LoadTopkV2

tengine-module/plugin/serializer/tensorflow/tf_serializer.cpp:4024–4053

}

static bool LoadTopkV2(TFNode* tf_node, TFGraph& tf_graph, StaticGraph* graph)
{
    StaticNode* node = tf_node->static_node;
    TFNode* input0 = tf_node->inputs[0];
    TFNode* input1 = tf_node->inputs[1];
    AddNodeInputTensor(node, input0->static_tensor);
    StaticTensor* tensor = input1->static_tensor;
    std::vector<int> dims;
    dims.push_back(1);
    SetTensorDim(tensor, dims);
    AddNodeInputTensor(node, input1->static_tensor);
    StaticOp* op = CreateStaticOp(graph, "TopKV2");
    TopKV2Param param = any_cast<TopKV2Param>(OpManager::GetOpDefParam("TopKV2"));

    const tensorflow::NodeDef* node_def = tf_node->pb_defs[0];
    tensorflow::AttrValue value0;
    GetAttrValue(node_def, "sorted", value0);
    param.sorted = value0.b();

    int* value = ( int* )LoadConstParam(input1);
    param.k = value[0];
    SetOperatorParam(op, param);
    SetNodeOp(node, op);

    StaticTensor* output1 = CreateStaticTensor(graph, "output1");
    SetTensorDataType(output1, DataType::GetTypeID("float32"));
    AddNodeOutputTensor(node, output1);

    return true;
}

static bool LoadFloor(TFNode* tf_node, TFGraph& tf_graph, StaticGraph* graph)
{