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/include/armnn/Types.hpp b/include/armnn/Types.hpp
index 36e3c5b..693a050 100644
--- a/include/armnn/Types.hpp
+++ b/include/armnn/Types.hpp
@@ -180,4 +180,9 @@
/// Define LayerGuid type.
using LayerGuid = unsigned int;
+class ITensorHandle;
+
+/// Define the type of callback for the debug layer to call
+using DebugCallbackFunction = std::function<void(LayerGuid, unsigned int, ITensorHandle*)>;
+
} // namespace armnn