aboutsummaryrefslogtreecommitdiff
path: root/src/backends/cl/workloads/ClConvertFp16ToFp32Workload.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/cl/workloads/ClConvertFp16ToFp32Workload.hpp')
-rw-r--r--src/backends/cl/workloads/ClConvertFp16ToFp32Workload.hpp5
1 files changed, 5 insertions, 0 deletions
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 <arm_compute/runtime/CL/functions/CLDepthConvertLayer.h>
+#include <cl/ICLTensorProxy.hpp>
+
namespace armnn
{
@@ -29,6 +31,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 ClConvertFp16ToFp32WorkloadValidate(const TensorInfo& input, const TensorInfo& output);