src/armnn/SubGraph.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72

//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//

#include "Layer.hpp"
#include "SubGraph.hpp"

#include <boost/numeric/conversion/cast.hpp>

namespace armnn
{

SubGraph::SubGraph()
{
}

SubGraph::SubGraph(InputSlots && inputs,
                   OutputSlots && outputs,
                   Layers && layers)
: m_InputSlots{inputs}
, m_OutputSlots{outputs}
, m_Layers{layers}
{
}

const SubGraph::InputSlots & SubGraph::GetInputSlots() const
{
    return m_InputSlots;
}

const SubGraph::OutputSlots & SubGraph::GetOutputSlots() const
{
    return m_OutputSlots;
}

const InputSlot* SubGraph::GetInputSlot(unsigned int index) const
{
    return m_InputSlots.at(index);
}

InputSlot* SubGraph::GetInputSlot(unsigned int index)
{
    return  m_InputSlots.at(index);
}

const OutputSlot* SubGraph::GetOutputSlot(unsigned int index) const
{
    return m_OutputSlots.at(index);
}

OutputSlot* SubGraph::GetOutputSlot(unsigned int index)
{
    return m_OutputSlots.at(index);
}

unsigned int SubGraph::GetNumInputSlots() const
{
    return boost::numeric_cast<unsigned int>(m_InputSlots.size());
}

unsigned int SubGraph::GetNumOutputSlots() const
{
    return boost::numeric_cast<unsigned int>(m_OutputSlots.size());
}

const SubGraph::Layers & SubGraph::GetLayers() const
{
    return m_Layers;
}

} // namespace armnn