diff options
Diffstat (limited to 'src/backends/neon/workloads/NeonMergerWorkload.hpp')
| -rw-r--r-- | src/backends/neon/workloads/NeonMergerWorkload.hpp | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/src/backends/neon/workloads/NeonMergerWorkload.hpp b/src/backends/neon/workloads/NeonMergerWorkload.hpp index 3432c626cb..1dd9309aff 100644 --- a/src/backends/neon/workloads/NeonMergerWorkload.hpp +++ b/src/backends/neon/workloads/NeonMergerWorkload.hpp @@ -9,7 +9,8 @@ #include <arm_compute/core/Error.h> #include <arm_compute/runtime/IFunction.h> -# +#include <arm_compute/runtime/NEON/functions/NEConcatenateLayer.h> + #include <memory> namespace armnn @@ -27,9 +28,7 @@ public: void Execute() const override; private: - std::unique_ptr<arm_compute::IFunction> m_Layer; - bool m_Execute; - + std::unique_ptr<arm_compute::NEConcatenateLayer> m_Layer; }; } //namespace armnn |