Function GetConnection

src/armnn/test/SubgraphViewTests.cpp:2063–2083

Workaround function used to get the original OutputSlot connected to the InputSlot of a SubgraphView As working copy SubgraphViews do not have connections on boundary it finds the corresponding InputSlot on the Original SubgraphView and then returns the OutputSlot connected to it. Using this function to test against the simpler API: SubgraphView::GetOriginalInputSlots().

// on the Original SubgraphView and then returns the OutputSlot connected to it.
// Using this function to test against the simpler API: SubgraphView::GetOriginalInputSlots().
const IOutputSlot* GetConnection(IInputSlot* inputSlot,
                                 const SubgraphView& workingCopy,
                                 const SubgraphView& original)
{
    const IOutputSlot* res = inputSlot->GetConnection();
    if (res)
    {
        return res;
    }

    const SubgraphView::IInputSlots& workingCopyInputSlots = workingCopy.GetIInputSlots();
    const SubgraphView::IInputSlots& originalInputSlots = original.GetIInputSlots();
    for (SubgraphView::InputSlots::size_type i = 0; i < workingCopyInputSlots.size(); i++)
    {
        if (workingCopyInputSlots[i] == inputSlot)
        {
            return originalInputSlots[i]->GetConnection();
        }
    }
    return nullptr;
}

// Workaround function used to get the original InputSlot connected to the OutputSlot of a SubgraphView
// As working copy SubgraphViews do not have connections on boundary it finds the corresponding OutputSlot