From a9a1cf117072b8aa81e27bab4d1d3e356d8ea51d Mon Sep 17 00:00:00 2001 From: Nattapat Chaimanowong Date: Mon, 3 Dec 2018 16:06:49 +0000 Subject: IVGCVSW-2315 Add DebugLayer and no-op factory method Change-Id: I5455b720565248ff94278e76887d63f8434a7b58 --- src/armnn/layers/DebugLayer.hpp | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) create mode 100644 src/armnn/layers/DebugLayer.hpp (limited to 'src/armnn/layers/DebugLayer.hpp') diff --git a/src/armnn/layers/DebugLayer.hpp b/src/armnn/layers/DebugLayer.hpp new file mode 100644 index 0000000000..39ef9c7ae0 --- /dev/null +++ b/src/armnn/layers/DebugLayer.hpp @@ -0,0 +1,27 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// +#pragma once + +#include + +namespace armnn +{ + +class DebugLayer : public Layer +{ +public: + virtual std::unique_ptr CreateWorkload(const Graph& graph, + const IWorkloadFactory& factory) const override; + + DebugLayer* Clone(Graph& graph) const override; + + void ValidateTensorShapesFromInputs() override; + +protected: + DebugLayer(const char* name); + ~DebugLayer() = default; +}; + +} // namespace armnn -- cgit v1.2.1