aboutsummaryrefslogtreecommitdiff
path: root/src/backends/cl/workloads/ClConvertFp32ToFp16Workload.hpp
diff options
context:
space:
mode:
authorJim Flynn <jim.flynn@arm.com>2022-02-14 11:22:29 +0000
committerJim Flynn <jim.flynn@arm.com>2022-02-14 14:14:20 +0000
commit5e7335b8ac8a63b81706316fda11348579a04726 (patch)
tree4ff8b7ece57d78a854420538df8a3c670cf15e95 /src/backends/cl/workloads/ClConvertFp32ToFp16Workload.hpp
parentfe1b067a677c0c991289ff2409e964b2b1b49318 (diff)
downloadarmnn-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.hpp5
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);