From 738c2e6a647b886750e1bc3daa6dd615a0244baa Mon Sep 17 00:00:00 2001
From: keidav01 <keith.davis@arm.com>
Date: Tue, 11 Dec 2018 16:14:20 +0000
Subject: IVGCVSW-1434 Add debug mode to Optimizer     * Modified optimizer to
 support debug mode via DebugLayer

Change-Id: Ic8f313778e55540c182cf99876c44a0823be04c6
---
 src/armnn/optimizations/AddDebug.hpp | 37 ++++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)
 create mode 100644 src/armnn/optimizations/AddDebug.hpp

(limited to 'src/armnn/optimizations/AddDebug.hpp')

diff --git a/src/armnn/optimizations/AddDebug.hpp b/src/armnn/optimizations/AddDebug.hpp
new file mode 100644
index 0000000000..60271b0d77
--- /dev/null
+++ b/src/armnn/optimizations/AddDebug.hpp
@@ -0,0 +1,37 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+#pragma once
+
+#include "Optimization.hpp"
+#include "NetworkUtils.hpp"
+
+namespace armnn
+{
+namespace optimizations
+{
+
+class AddDebugImpl
+{
+public:
+
+    void Run(Graph& graph, Layer& layer) const
+    {
+        if (layer.GetType() != LayerType::Debug && layer.GetType() != LayerType::Output)
+        {
+            // if the inputs/outputs of this layer do not have a debug layer
+            // insert the debug layer after them
+            InsertDebugLayerAfter(graph, layer);
+        }
+    }
+
+protected:
+    AddDebugImpl() = default;
+    ~AddDebugImpl() = default;
+};
+
+using InsertDebugLayer = OptimizeForType<Layer, AddDebugImpl>;
+
+} // namespace optimizations
+} // namespace armnn
-- 
cgit v1.2.1