diff options
Diffstat (limited to 'src/backends/neon/workloads/NeonMergerWorkload.hpp')
| -rw-r--r-- | src/backends/neon/workloads/NeonMergerWorkload.hpp | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/src/backends/neon/workloads/NeonMergerWorkload.hpp b/src/backends/neon/workloads/NeonMergerWorkload.hpp index a4f36d18bc..3432c626cb 100644 --- a/src/backends/neon/workloads/NeonMergerWorkload.hpp +++ b/src/backends/neon/workloads/NeonMergerWorkload.hpp @@ -6,7 +6,11 @@ #pragma once #include <backendsCommon/Workload.hpp> -#include <neon/workloads/NeonWorkloadUtils.hpp> + +#include <arm_compute/core/Error.h> +#include <arm_compute/runtime/IFunction.h> +# +#include <memory> namespace armnn { @@ -23,7 +27,7 @@ public: void Execute() const override; private: - mutable arm_compute::NEConcatenateLayer m_Layer; + std::unique_ptr<arm_compute::IFunction> m_Layer; bool m_Execute; }; |