aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/graph/IDeviceBackend.h
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/graph/IDeviceBackend.h')
-rw-r--r--arm_compute/graph/IDeviceBackend.h8
1 files changed, 7 insertions, 1 deletions
diff --git a/arm_compute/graph/IDeviceBackend.h b/arm_compute/graph/IDeviceBackend.h
index 358d26af81..cf54976c28 100644
--- a/arm_compute/graph/IDeviceBackend.h
+++ b/arm_compute/graph/IDeviceBackend.h
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2018 ARM Limited.
+ * Copyright (c) 2018-2019 ARM Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -28,6 +28,7 @@
#include "arm_compute/graph/Types.h"
#include "arm_compute/runtime/IFunction.h"
#include "arm_compute/runtime/IMemoryManager.h"
+#include "arm_compute/runtime/IWeightsManager.h"
#include <memory>
@@ -112,6 +113,11 @@ public:
* @return Memory manager
*/
virtual std::shared_ptr<arm_compute::IMemoryManager> create_memory_manager(MemoryManagerAffinity affinity) = 0;
+ /** Create a backend weights manager
+ *
+ * @return Weights manager
+ */
+ virtual std::shared_ptr<arm_compute::IWeightsManager> create_weights_manager() = 0;
};
} // namespace backends
} // namespace graph