diff options
| author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2018-10-11 10:29:15 +0100 |
|---|---|---|
| committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-22 16:57:53 +0100 |
| commit | a76698c34941ad5cf67fe114be05b038a31d98a7 (patch) | |
| tree | 482b56a4449cc259708f899356bfa0c1610b05aa /src/backends/cl/workloads/ClReshapeWorkload.hpp | |
| parent | ac9e096a574db91fbcc42c2ee919a1d1e57b7fd3 (diff) | |
| download | armnn-a76698c34941ad5cf67fe114be05b038a31d98a7.tar.gz | |
IVGCVSW-1951 Remove type templating from ClReshapeWorkload
Change-Id: I5349629bc5b36e5b5029a158bf888c09c3bda4b0
Diffstat (limited to 'src/backends/cl/workloads/ClReshapeWorkload.hpp')
| -rw-r--r-- | src/backends/cl/workloads/ClReshapeWorkload.hpp | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/src/backends/cl/workloads/ClReshapeWorkload.hpp b/src/backends/cl/workloads/ClReshapeWorkload.hpp new file mode 100644 index 0000000000..f949f764b2 --- /dev/null +++ b/src/backends/cl/workloads/ClReshapeWorkload.hpp @@ -0,0 +1,26 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#pragma once + +#include <backends/Workload.hpp> + +#include <arm_compute/runtime/CL/CLFunctions.h> + +namespace armnn +{ + +class ClReshapeWorkload : public BaseWorkload<ReshapeQueueDescriptor> +{ +public: + ClReshapeWorkload(const ReshapeQueueDescriptor& descriptor, const WorkloadInfo& info); + + void Execute() const override; + +private: + mutable arm_compute::CLReshapeLayer m_Layer; +}; + +} //namespace armnn |