aboutsummaryrefslogtreecommitdiff
path: root/src/backends/cl
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/cl')
-rw-r--r--src/backends/cl/test/ClOptimizedNetworkTests.cpp3
-rw-r--r--src/backends/cl/test/Fp16SupportTest.cpp1
-rw-r--r--src/backends/cl/workloads/ClLstmFloatWorkload.hpp2
-rw-r--r--src/backends/cl/workloads/ClQuantizedLstmWorkload.hpp3
4 files changed, 5 insertions, 4 deletions
diff --git a/src/backends/cl/test/ClOptimizedNetworkTests.cpp b/src/backends/cl/test/ClOptimizedNetworkTests.cpp
index 824375ef86..c2a8005e8a 100644
--- a/src/backends/cl/test/ClOptimizedNetworkTests.cpp
+++ b/src/backends/cl/test/ClOptimizedNetworkTests.cpp
@@ -5,7 +5,6 @@
#include "ClWorkloadFactoryHelper.hpp"
-#include <armnn/ArmNN.hpp>
#include <Network.hpp>
#include <test/GraphUtils.hpp>
@@ -101,4 +100,4 @@ BOOST_AUTO_TEST_CASE(FP16TurboModeTestOnGpuAcc)
BOOST_TEST(GraphHasNamedLayer(graph, "output layer"));
}
-BOOST_AUTO_TEST_SUITE_END(); \ No newline at end of file
+BOOST_AUTO_TEST_SUITE_END();
diff --git a/src/backends/cl/test/Fp16SupportTest.cpp b/src/backends/cl/test/Fp16SupportTest.cpp
index f117c92e49..ee5163f668 100644
--- a/src/backends/cl/test/Fp16SupportTest.cpp
+++ b/src/backends/cl/test/Fp16SupportTest.cpp
@@ -3,7 +3,6 @@
// SPDX-License-Identifier: MIT
//
-#include <armnn/ArmNN.hpp>
#include <armnn/Descriptors.hpp>
#include <armnn/IRuntime.hpp>
#include <armnn/INetwork.hpp>
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
{