diff options
Diffstat (limited to 'src/armnn/test/TestUtils.hpp')
-rw-r--r-- | src/armnn/test/TestUtils.hpp | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/src/armnn/test/TestUtils.hpp b/src/armnn/test/TestUtils.hpp index 9b1a3bf18d..9c5f672a5a 100644 --- a/src/armnn/test/TestUtils.hpp +++ b/src/armnn/test/TestUtils.hpp @@ -7,6 +7,7 @@ #include <armnn/INetwork.hpp> #include <Graph.hpp> +#include <Runtime.hpp> void Connect(armnn::IConnectableLayer* from, armnn::IConnectableLayer* to, const armnn::TensorInfo& tensorInfo, unsigned int fromIndex = 0, unsigned int toIndex = 0); @@ -47,3 +48,10 @@ bool CheckRelatedLayers(armnn::Graph& graph, const std::list<std::string>& testR return true; } + +namespace armnn +{ + +profiling::ProfilingService& GetProfilingService(armnn::Runtime* runtime); + +} // namespace armnn
\ No newline at end of file |