diff options
| author | jimfly01 <jim.flynn@arm.com> | 2018-09-21 12:09:51 +0100 |
|---|---|---|
| committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-10 16:16:57 +0100 |
| commit | e1fa50c70c647e3ba6dec0eb49ce82a3820fb6bb (patch) | |
| tree | 6cc31b7a37e3ab81a069d124bcead4fe1ad97119 /src/backends/ClWorkloads/ClPadWorkload.hpp | |
| parent | 1e4c31dafb1c8984a126fa1d211ed8f9eedaf7cc (diff) | |
| download | armnn-e1fa50c70c647e3ba6dec0eb49ce82a3820fb6bb.tar.gz | |
IVGCVSW-1882: Added ClPadWorkload
* added ClPadWorkload class
* Updated ClWorkloadFactory
* added IsPadSupportedCl function
* Updated build files
Change-Id: Iadccc340a0eae1ff30fab6441854acac7f8edc92
Diffstat (limited to 'src/backends/ClWorkloads/ClPadWorkload.hpp')
| -rw-r--r-- | src/backends/ClWorkloads/ClPadWorkload.hpp | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/src/backends/ClWorkloads/ClPadWorkload.hpp b/src/backends/ClWorkloads/ClPadWorkload.hpp new file mode 100644 index 0000000000..0ec560d545 --- /dev/null +++ b/src/backends/ClWorkloads/ClPadWorkload.hpp @@ -0,0 +1,32 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#pragma once + +#include "backends/WorkloadData.hpp" +#include "backends/Workload.hpp" +#include <arm_compute/runtime/CL/functions/CLPadLayer.h> + +namespace armnn { + +template <armnn::DataType... dataTypes> +class ClPadWorkload : public TypedWorkload<PadQueueDescriptor, dataTypes...> +{ +public: + ClPadWorkload(const PadQueueDescriptor& descriptor, const WorkloadInfo& info); + + void Execute() const override; + +private: + mutable arm_compute::CLPadLayer m_Layer; +}; + +bool ClPadValidate(const TensorInfo& input, + const TensorInfo& output, + const PadDescriptor& descriptor, + std::string* reasonIfUnsupported); + +} //namespace armnn + |