From c577f2c6a3b4ddb6ba87a882723c53a248afbeba Mon Sep 17 00:00:00 2001 From: telsoa01 Date: Fri, 31 Aug 2018 09:22:23 +0100 Subject: Release 18.08 --- src/armnn/LoadedNetwork.hpp | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) (limited to 'src/armnn/LoadedNetwork.hpp') diff --git a/src/armnn/LoadedNetwork.hpp b/src/armnn/LoadedNetwork.hpp index 79a0b267e9..286f804234 100644 --- a/src/armnn/LoadedNetwork.hpp +++ b/src/armnn/LoadedNetwork.hpp @@ -8,6 +8,7 @@ #include "armnn/Types.hpp" #include "Network.hpp" #include "LayerFwd.hpp" +#include "Profiling.hpp" #include "backends/RefWorkloadFactory.hpp" #include "backends/NeonWorkloadFactory.hpp" #include "backends/ClWorkloadFactory.hpp" @@ -33,10 +34,15 @@ public: Status EnqueueWorkload(const InputTensors& inputTensors, const OutputTensors& outputTensors); static std::unique_ptr MakeLoadedNetwork(std::unique_ptr net, - bool useCpuRefAsFallback); + std::string & errorMessage); + + // NOTE we return by reference as the purpose of this method is only to provide + // access to the private m_Profiler and in theory we should not need to increment + // the shared_ptr's reference counter + const std::shared_ptr& GetProfiler() const { return m_Profiler; } private: - LoadedNetwork(std::unique_ptr net, bool useCpuRefAsFallback); + LoadedNetwork(std::unique_ptr net); void EnqueueInput(const BindableLayer& layer, ITensorHandle* tensorHandle, const TensorInfo& tensorInfo); @@ -54,6 +60,7 @@ private: std::unique_ptr m_OptimizedNetwork; std::vector< std::unique_ptr > m_WorkloadQueue; + std::shared_ptr m_Profiler; }; } -- cgit v1.2.1