diff options
author | Derek Lamberti <derek.lamberti@arm.com> | 2021-10-13 14:32:12 +0100 |
---|---|---|
committer | Derek Lamberti <derek.lamberti@arm.com> | 2021-10-15 15:29:43 +0100 |
commit | e155bbf0e9be6b4d7974297585a59207cd89b00a (patch) | |
tree | 78f5d5f4f2a3a8551854fdd46eac7f599e371bd1 /src/armnn/Graph.cpp | |
parent | 827e4bf519ec1e4af2fb0b32d476b8b1c73b877c (diff) | |
download | armnn-e155bbf0e9be6b4d7974297585a59207cd89b00a.tar.gz |
Refactor: Profiler moved to Graph
* This is to enable later work to instrument the Optimizer.
Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>
Change-Id: I2cf1fe022e0d100d6d8705adfbb8cab3ffc96a86
Diffstat (limited to 'src/armnn/Graph.cpp')
-rw-r--r-- | src/armnn/Graph.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/armnn/Graph.cpp b/src/armnn/Graph.cpp index ebfc829340..30639b12e8 100644 --- a/src/armnn/Graph.cpp +++ b/src/armnn/Graph.cpp @@ -27,6 +27,7 @@ namespace armnn Graph::Graph(const Graph& other) : m_LayersInOrder(other.m_LayersInOrder) +, m_Profiler(other.m_Profiler) { std::unordered_map<const Layer*, Layer*> otherToClonedMap; @@ -636,4 +637,9 @@ void Graph::ConstructErrorMessageForUnconnectedInputs(Layer* const layer, throw LayerValidationException(message.str()); } +const std::shared_ptr<IProfiler>& Graph::GetProfiler() const +{ + return m_Profiler; +} + } // namespace armnn |