diff options
author | David Beck <david.beck@arm.com> | 2018-10-26 16:03:03 +0100 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2018-10-26 16:40:49 +0000 |
commit | f98d21a244c384c5942e4b261d168f65ecc8a797 (patch) | |
tree | 65de8992145a7c2801d9b626f61f2ec6ebbf7003 /src/armnn/SubGraph.cpp | |
parent | 5f70318b08907eb1612dbe88d53857110909cb42 (diff) | |
download | armnn-f98d21a244c384c5942e4b261d168f65ecc8a797.tar.gz |
IVGCVSW-1896 : SubGraph selector
Change-Id: Iae9a76b10d84d7ba80136b54355f8d37d2df475d
Diffstat (limited to 'src/armnn/SubGraph.cpp')
-rw-r--r-- | src/armnn/SubGraph.cpp | 72 |
1 files changed, 72 insertions, 0 deletions
diff --git a/src/armnn/SubGraph.cpp b/src/armnn/SubGraph.cpp new file mode 100644 index 0000000000..5d41f32932 --- /dev/null +++ b/src/armnn/SubGraph.cpp @@ -0,0 +1,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 |