aboutsummaryrefslogtreecommitdiff
path: root/src/backends/backendsCommon/TensorHandleFactoryRegistry.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/backendsCommon/TensorHandleFactoryRegistry.hpp')
-rw-r--r--src/backends/backendsCommon/TensorHandleFactoryRegistry.hpp5
1 files changed, 3 insertions, 2 deletions
diff --git a/src/backends/backendsCommon/TensorHandleFactoryRegistry.hpp b/src/backends/backendsCommon/TensorHandleFactoryRegistry.hpp
index 2a7c6f36d9..434b64ffaf 100644
--- a/src/backends/backendsCommon/TensorHandleFactoryRegistry.hpp
+++ b/src/backends/backendsCommon/TensorHandleFactoryRegistry.hpp
@@ -1,12 +1,13 @@
//
-// Copyright © 2017 Arm Ltd. All rights reserved.
+// Copyright © 2017,2022 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//
#pragma once
+#include <armnn/Types.hpp>
#include <armnn/backends/ITensorHandleFactory.hpp>
-
+#include <map>
#include <memory>
#include <vector>