From 81e601c5a5ebf3de3dd6418942708158de50252a Mon Sep 17 00:00:00 2001 From: Matthew Jackson Date: Thu, 11 Jul 2019 12:07:09 +0100 Subject: IVGCVSW-3419 Add reference workload support for the new Stack layer * Added reference workload for the Stack layer * Added factory methods * Added validation support * Added unit tests Signed-off-by: Matthew Jackson Change-Id: Ib14b72c15f53a2a2ca152afc357ce2aa405ccc88 --- .../reference/workloads/RefStackWorkload.hpp | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) create mode 100644 src/backends/reference/workloads/RefStackWorkload.hpp (limited to 'src/backends/reference/workloads/RefStackWorkload.hpp') diff --git a/src/backends/reference/workloads/RefStackWorkload.hpp b/src/backends/reference/workloads/RefStackWorkload.hpp new file mode 100644 index 0000000000..ceb27d9f60 --- /dev/null +++ b/src/backends/reference/workloads/RefStackWorkload.hpp @@ -0,0 +1,22 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#pragma once + +#include +#include + +namespace armnn +{ + +class RefStackWorkload : public BaseWorkload +{ +public: + explicit RefStackWorkload(const StackQueueDescriptor& descriptor, + const WorkloadInfo& info); + virtual void Execute() const override; +}; + +} // namespace armnn -- cgit v1.2.1