diff options
author | Jim Flynn <jim.flynn@arm.com> | 2022-02-14 11:22:29 +0000 |
---|---|---|
committer | Colm Donelan <colm.donelan@arm.com> | 2022-02-16 21:36:34 +0000 |
commit | a00bad1140223453e72a134388b209f9fa296d00 (patch) | |
tree | 81774e9748b0726981fc3160d1fd232bc6359512 /src/backends/cl/workloads/ClConvertFp32ToFp16Workload.hpp | |
parent | 01055eba0287778f509ac292aa6fa068980e2034 (diff) | |
download | armnn-a00bad1140223453e72a134388b209f9fa296d00.tar.gz |
Add implementation of reconfigure function to Cl Convert workloads
Change-Id: Id993a300847bde012dd38ef7f44fcd3c14867349
Signed-off-by: Jim Flynn <jim.flynn@arm.com>
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Diffstat (limited to 'src/backends/cl/workloads/ClConvertFp32ToFp16Workload.hpp')
-rw-r--r-- | src/backends/cl/workloads/ClConvertFp32ToFp16Workload.hpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/backends/cl/workloads/ClConvertFp32ToFp16Workload.hpp b/src/backends/cl/workloads/ClConvertFp32ToFp16Workload.hpp index a4baef8555..adffd8123e 100644 --- a/src/backends/cl/workloads/ClConvertFp32ToFp16Workload.hpp +++ b/src/backends/cl/workloads/ClConvertFp32ToFp16Workload.hpp @@ -9,6 +9,8 @@ #include <arm_compute/runtime/CL/functions/CLDepthConvertLayer.h> +#include <cl/ICLTensorProxy.hpp> + namespace armnn { @@ -28,6 +30,9 @@ public: private: mutable arm_compute::CLDepthConvertLayer m_Layer; virtual void Reconfigure(); + + std::unique_ptr<ICLTensorProxy> m_InputProxy; + std::unique_ptr<ICLTensorProxy> m_OutputProxy; }; arm_compute::Status ClConvertFp32ToFp16WorkloadValidate(const TensorInfo& input, const TensorInfo& output); |