Function FindTensor

core/lib/static_graph.cpp:146–159

}

StaticTensor* FindTensor(StaticGraph* graph, const std::string& tensor_name)
{
    int tensor_number = graph->tensor_list.size();

    for(int i = 0; i < tensor_number; i++)
    {
        StaticTensor* tensor = graph->tensor_list[i].get();

        if(tensor->name == tensor_name)
            return tensor;
    }

    return nullptr;
}

StaticTensor* FindConstTensor(StaticGraph* graph, const std::string& tensor_name)
{