diff options
Diffstat (limited to 'src/backends/reference/workloads/RefLstmWorkload.hpp')
-rw-r--r-- | src/backends/reference/workloads/RefLstmWorkload.hpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/backends/reference/workloads/RefLstmWorkload.hpp b/src/backends/reference/workloads/RefLstmWorkload.hpp index 72f6360281..57526c9ba2 100644 --- a/src/backends/reference/workloads/RefLstmWorkload.hpp +++ b/src/backends/reference/workloads/RefLstmWorkload.hpp @@ -7,13 +7,13 @@ #include <armnn/TypesUtils.hpp> -#include <armnn/backends/Workload.hpp> +#include "RefBaseWorkload.hpp" #include <armnn/backends/WorkloadData.hpp> namespace armnn { -class RefLstmWorkload : public BaseWorkload<LstmQueueDescriptor> +class RefLstmWorkload : public RefBaseWorkload<LstmQueueDescriptor> { public: explicit RefLstmWorkload(const LstmQueueDescriptor& descriptor, const WorkloadInfo& info); |