diff options
Diffstat (limited to 'src/armnn/LoadedNetwork.hpp')
-rw-r--r-- | src/armnn/LoadedNetwork.hpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/armnn/LoadedNetwork.hpp b/src/armnn/LoadedNetwork.hpp index 3a44206683..c7dd37fdea 100644 --- a/src/armnn/LoadedNetwork.hpp +++ b/src/armnn/LoadedNetwork.hpp @@ -42,7 +42,7 @@ public: Status EnqueueWorkload(const InputTensors& inputTensors, const OutputTensors& outputTensors); - static std::unique_ptr<LoadedNetwork> MakeLoadedNetwork(std::unique_ptr<OptimizedNetwork> net, + static std::unique_ptr<LoadedNetwork> MakeLoadedNetwork(std::unique_ptr<IOptimizedNetwork> net, std::string & errorMessage, const INetworkProperties& networkProperties, profiling::ProfilingService& profilingService); @@ -63,7 +63,7 @@ public: private: void AllocateWorkingMemory(std::lock_guard<std::mutex>& lock); - LoadedNetwork(std::unique_ptr<OptimizedNetwork> net, + LoadedNetwork(std::unique_ptr<IOptimizedNetwork> net, const INetworkProperties& networkProperties, profiling::ProfilingService& profilingService); @@ -87,7 +87,7 @@ private: BackendPtrMap m_Backends; WorkloadFactoryMap m_WorkloadFactories; - std::unique_ptr<OptimizedNetwork> m_OptimizedNetwork; + std::unique_ptr<IOptimizedNetwork> m_OptimizedNetwork; WorkloadQueue m_InputQueue; WorkloadQueue m_WorkloadQueue; WorkloadQueue m_OutputQueue; |