From 5e7335b8ac8a63b81706316fda11348579a04726 Mon Sep 17 00:00:00 2001 From: Jim Flynn Date: Mon, 14 Feb 2022 11:22:29 +0000 Subject: Add implementation of reconfigure function to Cl Convert workloads Change-Id: Id993a300847bde012dd38ef7f44fcd3c14867349 Signed-off-by: Jim Flynn Signed-off-by: Narumol Prangnawarat --- src/backends/cl/workloads/ClConvertFp16ToFp32Workload.hpp | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'src/backends/cl/workloads/ClConvertFp16ToFp32Workload.hpp') diff --git a/src/backends/cl/workloads/ClConvertFp16ToFp32Workload.hpp b/src/backends/cl/workloads/ClConvertFp16ToFp32Workload.hpp index e78ffa4aff..efca378f97 100644 --- a/src/backends/cl/workloads/ClConvertFp16ToFp32Workload.hpp +++ b/src/backends/cl/workloads/ClConvertFp16ToFp32Workload.hpp @@ -9,6 +9,8 @@ #include +#include + namespace armnn { @@ -29,6 +31,9 @@ public: private: mutable arm_compute::CLDepthConvertLayer m_Layer; virtual void Reconfigure(); + + std::unique_ptr m_InputProxy; + std::unique_ptr m_OutputProxy; }; arm_compute::Status ClConvertFp16ToFp32WorkloadValidate(const TensorInfo& input, const TensorInfo& output); -- cgit v1.2.1