Method GetInputTensor

core/lib/graph.cpp:949–965

}

Tensor* Graph::GetInputTensor(const std::string& name)
{
    for(unsigned int i = 0; i < input_nodes.size(); i++)
    {
        Node* node = input_nodes[i];

        for(unsigned int j = 0; j < node->GetInputNum(); j++)
        {
            Tensor* tensor = node->GetInputTensor(j);

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

    return nullptr;
}

Tensor* Graph::GetOutputTensor(const std::string& name)
{