From 39ef3e595b8e0158875c0346855af5af679d78ad Mon Sep 17 00:00:00 2001
From: Matthew Bentham <Matthew.Bentham@arm.com>
Date: Mon, 20 Jan 2020 10:09:09 +0000
Subject: Remove use of ArmNN.hpp where it is unnecessary.

Signed-off-by: Matthew Bentham <Matthew.Bentham@arm.com>
Change-Id: I6c90eab1c6bad76d5c7b3bf094998fa4b454bcd1
---
 src/backends/cl/workloads/ClLstmFloatWorkload.hpp     | 2 ++
 src/backends/cl/workloads/ClQuantizedLstmWorkload.hpp | 3 ++-
 2 files changed, 4 insertions(+), 1 deletion(-)

(limited to 'src/backends/cl/workloads')

diff --git a/src/backends/cl/workloads/ClLstmFloatWorkload.hpp b/src/backends/cl/workloads/ClLstmFloatWorkload.hpp
index 5bd67c256f..ae42d6ecc3 100644
--- a/src/backends/cl/workloads/ClLstmFloatWorkload.hpp
+++ b/src/backends/cl/workloads/ClLstmFloatWorkload.hpp
@@ -5,6 +5,8 @@
 
 #pragma once
 
+#include <armnn/Descriptors.hpp>
+#include <armnn/LstmParams.hpp>
 #include <backendsCommon/Workload.hpp>
 #include <backendsCommon/WorkloadData.hpp>
 
diff --git a/src/backends/cl/workloads/ClQuantizedLstmWorkload.hpp b/src/backends/cl/workloads/ClQuantizedLstmWorkload.hpp
index c7d83755c7..580db490d6 100644
--- a/src/backends/cl/workloads/ClQuantizedLstmWorkload.hpp
+++ b/src/backends/cl/workloads/ClQuantizedLstmWorkload.hpp
@@ -5,10 +5,11 @@
 
 #pragma once
 
+#include <armnn/QuantizedLstmParams.hpp>
 #include <backendsCommon/Workload.hpp>
 #include <backendsCommon/WorkloadData.hpp>
 
-#include <arm_compute/runtime/CL/CLFunctions.h>
+#include <arm_compute/runtime/CL/functions/CLLSTMLayerQuantized.h>
 
 namespace armnn
 {
-- 
cgit v1.2.1