IVGCVSW-2866 Implement RegisterDebugCallback for RefDebugWorkload
Change-Id: I9144fb6b7d05561b5b8fd9db5dbe31c9257f10ca
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
diff --git a/include/armnn/Types.hpp b/include/armnn/Types.hpp
index bdddd37..438b6be 100644
--- a/include/armnn/Types.hpp
+++ b/include/armnn/Types.hpp
@@ -183,7 +183,10 @@
class ITensorHandle;
-/// Define the type of callback for the debug layer to call
-using DebugCallbackFunction = std::function<void(LayerGuid, unsigned int, ITensorHandle*)>;
+/// Define the type of callback for the Debug layer to call
+/// @param guid - guid of layer connected to the input of the Debug layer
+/// @param slotIndex - index of the output slot connected to the input of the Debug layer
+/// @param tensorHandle - TensorHandle for the input tensor to the Debug layer
+using DebugCallbackFunction = std::function<void(LayerGuid guid, unsigned int slotIndex, ITensorHandle* tensorHandle)>;
} // namespace armnn