diff options
Diffstat (limited to 'include/armnn/INetwork.hpp')
-rw-r--r-- | include/armnn/INetwork.hpp | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/include/armnn/INetwork.hpp b/include/armnn/INetwork.hpp index bceb07405a..2db6d5de83 100644 --- a/include/armnn/INetwork.hpp +++ b/include/armnn/INetwork.hpp @@ -704,6 +704,12 @@ protected: std::unique_ptr<NetworkImpl> pNetworkImpl; }; +namespace experimental +{ +class AsyncNetwork; +class WorkingMemHandle; +} + struct BackendSettings; struct OptimizationResult; class OptimizedNetworkImpl; @@ -723,6 +729,10 @@ public: protected: friend class LoadedNetwork; + + friend class experimental::AsyncNetwork; + friend class experimental::WorkingMemHandle; + friend Graph& GetGraphForTesting(IOptimizedNetwork* optNetPtr); friend ModelOptions& GetModelOptionsForTesting(IOptimizedNetwork* optNetPtr); friend IOptimizedNetworkPtr Optimize(const INetwork& inNetwork, |