IVGCVSW-2865 Extend IRuntime to add a new method RegisterDebugCallback(...)
* Made changes to LoadedNetwork and IWorkload to pass on the registered
callback function
Change-Id: I6ea10f2a299d6de8bf681c8ff36d3fbed1d6d887
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
diff --git a/src/armnn/Runtime.cpp b/src/armnn/Runtime.cpp
index 09be92c..f8b2462 100644
--- a/src/armnn/Runtime.cpp
+++ b/src/armnn/Runtime.cpp
@@ -231,4 +231,10 @@
return loadedNetwork->EnqueueWorkload(inputTensors, outputTensors);
}
+void Runtime::RegisterDebugCallback(NetworkId networkId, const DebugCallbackFunction& func)
+{
+ LoadedNetwork* loadedNetwork = GetLoadedNetworkPtr(networkId);
+ loadedNetwork->RegisterDebugCallback(func);
+}
+
}