diff options
Diffstat (limited to 'arm_compute/graph/IDeviceBackend.h')
-rw-r--r-- | arm_compute/graph/IDeviceBackend.h | 8 |
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 |