diff options
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 |