diff options
author | Mike Kelly <mike.kelly@arm.com> | 2018-11-28 11:52:08 +0000 |
---|---|---|
committer | Mike Kelly <mike.kelly@arm.com> | 2018-11-28 11:52:08 +0000 |
commit | 831faedd5d2f6306b23202fa4e450c6d241a44a0 (patch) | |
tree | ad025f07969618a054144aa45333209399656cb5 /src/backends/cl/workloads/ClBatchToSpaceNdWorkload.hpp | |
parent | 7e7261ed9a4f8d48c86e57044312e3df82189713 (diff) | |
download | armnn-831faedd5d2f6306b23202fa4e450c6d241a44a0.tar.gz |
IVGCVSW-1199: BATCH_TO_SPACE_ND integrate Arm Compute CL
Change-Id: Ic772acf481caed6e8a6b99f68a63dfc2a34c24dc
Diffstat (limited to 'src/backends/cl/workloads/ClBatchToSpaceNdWorkload.hpp')
-rw-r--r-- | src/backends/cl/workloads/ClBatchToSpaceNdWorkload.hpp | 30 |
1 files changed, 30 insertions, 0 deletions
diff --git a/src/backends/cl/workloads/ClBatchToSpaceNdWorkload.hpp b/src/backends/cl/workloads/ClBatchToSpaceNdWorkload.hpp new file mode 100644 index 0000000000..4db84a2787 --- /dev/null +++ b/src/backends/cl/workloads/ClBatchToSpaceNdWorkload.hpp @@ -0,0 +1,30 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#pragma once + +#include <backendsCommon/Workload.hpp> +#include <arm_compute/runtime/CL/CLFunctions.h> + +namespace armnn +{ + +arm_compute::Status ClBatchToSpaceNdWorkloadValidate(const TensorInfo& input, + const TensorInfo& output, + const BatchToSpaceNdDescriptor& desc); + +class ClBatchToSpaceNdWorkload : public BaseWorkload<BatchToSpaceNdQueueDescriptor> +{ +public: + ClBatchToSpaceNdWorkload(const BatchToSpaceNdQueueDescriptor& descriptor, const WorkloadInfo& info); + + void Execute() const override; + +private: + + mutable arm_compute::CLBatchToSpaceLayer m_Layer; +}; + +} //namespace armnn |