diff options
author | Jim Flynn <jim.flynn@arm.com> | 2022-02-14 11:22:29 +0000 |
---|---|---|
committer | Jim Flynn <jim.flynn@arm.com> | 2022-02-14 14:14:20 +0000 |
commit | 5e7335b8ac8a63b81706316fda11348579a04726 (patch) | |
tree | 4ff8b7ece57d78a854420538df8a3c670cf15e95 /src/backends/cl/workloads/ClConvertFp32ToFp16Workload.hpp | |
parent | fe1b067a677c0c991289ff2409e964b2b1b49318 (diff) | |
download | armnn-5e7335b8ac8a63b81706316fda11348579a04726.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); |