From ac5aa1faef759e704d369e8164ea91c6d7049946 Mon Sep 17 00:00:00 2001
From: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
Date: Wed, 5 Dec 2018 15:17:18 +0000
Subject: IVGCVSW-2315 Add descriptor for Debug

Change-Id: I6b26c681989ef815752186f027a245ef8da7aea2
---
 src/armnn/layers/DebugLayer.cpp | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

(limited to 'src/armnn/layers/DebugLayer.cpp')

diff --git a/src/armnn/layers/DebugLayer.cpp b/src/armnn/layers/DebugLayer.cpp
index c407643742..e83b17ee64 100644
--- a/src/armnn/layers/DebugLayer.cpp
+++ b/src/armnn/layers/DebugLayer.cpp
@@ -12,20 +12,23 @@
 namespace armnn
 {
 
-DebugLayer::DebugLayer(const char* name) : Layer(1, 1, LayerType::Debug, name)
+DebugLayer::DebugLayer(const DebugDescriptor& param, const char* name)
+    : LayerWithParameters(1, 1, LayerType::Debug, param, name)
 {}
 
 std::unique_ptr<IWorkload> DebugLayer::CreateWorkload(const Graph& graph,
                                                       const IWorkloadFactory& factory) const
 {
     DebugQueueDescriptor descriptor;
+    descriptor.m_Parameters.m_LayerName = m_Param.m_LayerName;
+    descriptor.m_Parameters.m_SlotIndex = m_Param.m_SlotIndex;
 
     return factory.CreateDebug(descriptor, PrepInfoAndDesc(descriptor, graph));
 }
 
 DebugLayer* DebugLayer::Clone(Graph& graph) const
 {
-    return CloneBase<DebugLayer>(graph, GetName());
+    return CloneBase<DebugLayer>(graph, m_Param, GetName());
 }
 
 void DebugLayer::ValidateTensorShapesFromInputs()
-- 
cgit v1.2.1