From 73d3e2e1616ba5dcdb0a190afba2463742bd4fcc Mon Sep 17 00:00:00 2001 From: Francis Murtagh Date: Thu, 29 Apr 2021 14:23:04 +0100 Subject: IVGCVSW-5819 5820 5821 Add MemorySourceFlags to TensorHandleFactoryRegistry::GetFactory * Modify Layer::CreateTensorHandles to include MemorySource * Modify INetworkProperties to add MemorySource * Disable Neon/Cl fallback tests until full import implementation complete Change-Id: Ia4fff6ea3d4bf6afca33aae358125ccaec7f9a38 Signed-off-by: Francis Murtagh --- src/backends/backendsCommon/TensorHandleFactoryRegistry.hpp | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'src/backends/backendsCommon/TensorHandleFactoryRegistry.hpp') diff --git a/src/backends/backendsCommon/TensorHandleFactoryRegistry.hpp b/src/backends/backendsCommon/TensorHandleFactoryRegistry.hpp index e9e76e73a6..525db56216 100644 --- a/src/backends/backendsCommon/TensorHandleFactoryRegistry.hpp +++ b/src/backends/backendsCommon/TensorHandleFactoryRegistry.hpp @@ -35,6 +35,10 @@ public: /// Returns nullptr if not found ITensorHandleFactory* GetFactory(ITensorHandleFactory::FactoryId id) const; + /// Overload of above allowing specification of Memory Source + ITensorHandleFactory* GetFactory(ITensorHandleFactory::FactoryId id, + MemorySource memSource) const; + /// Aquire memory required for inference void AquireMemory(); -- cgit v1.2.1