diff options
Diffstat (limited to 'src/backends/neon')
-rw-r--r-- | src/backends/neon/workloads/NeonLstmFloatWorkload.hpp | 2 | ||||
-rw-r--r-- | src/backends/neon/workloads/NeonQuantizedLstmWorkload.hpp | 1 |
2 files changed, 3 insertions, 0 deletions
diff --git a/src/backends/neon/workloads/NeonLstmFloatWorkload.hpp b/src/backends/neon/workloads/NeonLstmFloatWorkload.hpp index c116cdd967..9a9a878574 100644 --- a/src/backends/neon/workloads/NeonLstmFloatWorkload.hpp +++ b/src/backends/neon/workloads/NeonLstmFloatWorkload.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/neon/workloads/NeonQuantizedLstmWorkload.hpp b/src/backends/neon/workloads/NeonQuantizedLstmWorkload.hpp index c3bcf785ad..740cb4a316 100644 --- a/src/backends/neon/workloads/NeonQuantizedLstmWorkload.hpp +++ b/src/backends/neon/workloads/NeonQuantizedLstmWorkload.hpp @@ -5,6 +5,7 @@ #pragma once +#include <armnn/QuantizedLstmParams.hpp> #include <backendsCommon/Workload.hpp> #include <backendsCommon/WorkloadData.hpp> |