diff options
Diffstat (limited to 'src/backends/backendsCommon')
-rw-r--r-- | src/backends/backendsCommon/TensorHandleFactoryRegistry.hpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/TensorHandleFactoryRegistry.hpp b/src/backends/backendsCommon/TensorHandleFactoryRegistry.hpp index f926478432..e9e76e73a6 100644 --- a/src/backends/backendsCommon/TensorHandleFactoryRegistry.hpp +++ b/src/backends/backendsCommon/TensorHandleFactoryRegistry.hpp @@ -41,6 +41,11 @@ public: /// Release memory required for inference void ReleaseMemory(); + std::vector<std::shared_ptr<IMemoryManager>>& GetMemoryManagers() + { + return m_MemoryManagers; + } + private: std::vector<std::unique_ptr<ITensorHandleFactory>> m_Factories; std::vector<std::shared_ptr<IMemoryManager>> m_MemoryManagers; |