TensorHandleFactoryRegistry Class Reference

#include <TensorHandleFactoryRegistry.hpp>

Public Member Functions
	TensorHandleFactoryRegistry ()=default
	TensorHandleFactoryRegistry (const TensorHandleFactoryRegistry &other)=delete
	TensorHandleFactoryRegistry (TensorHandleFactoryRegistry &&other)=delete
void	RegisterFactory (std::unique_ptr< ITensorHandleFactory > allocator)
	Register a TensorHandleFactory and transfer ownership. More...
void	RegisterMemoryManager (std::shared_ptr< IMemoryManager > memoryManger)
	Register a memory manager with shared ownership. More...
ITensorHandleFactory *	GetFactory (ITensorHandleFactory::FactoryId id) const
	Find a TensorHandleFactory by Id Returns nullptr if not found. More...
void	AquireMemory ()
	Aquire memory required for inference. More...
void	ReleaseMemory ()
	Release memory required for inference. More...

Detailed Description

Definition at line 20 of file TensorHandleFactoryRegistry.hpp.

Constructor & Destructor Documentation

TensorHandleFactoryRegistry() [1/3]

TensorHandleFactoryRegistry ( )

default

TensorHandleFactoryRegistry() [2/3]

TensorHandleFactoryRegistry ( const TensorHandleFactoryRegistry &  other )

delete

TensorHandleFactoryRegistry() [3/3]

TensorHandleFactoryRegistry ( TensorHandleFactoryRegistry &&  other )

delete

Member Function Documentation

AquireMemory()

void AquireMemory

(

)

Aquire memory required for inference.

Definition at line 52 of file TensorHandleFactoryRegistry.cpp.

{
    for (auto& mgr : m_MemoryManagers)
    {
        mgr->Acquire();
    }
}

GetFactory()

ITensorHandleFactory * GetFactory

(

ITensorHandleFactory::FactoryId

id

)

const

Find a TensorHandleFactory by Id Returns nullptr if not found.

Definition at line 39 of file TensorHandleFactoryRegistry.cpp.

Referenced by Graph::AddCompatibilityLayers(), armnn::CalculateEdgeStrategy(), armnn::CalculateSlotOption(), armnn::CalculateSlotOptionForInput(), ConcatLayer::CreateTensorHandles(), SplitterLayer::CreateTensorHandles(), Layer::CreateTensorHandles(), SplitterLayer::CreateWorkload(), ConcatLayer::CreateWorkload(), and armnn::RequiresCopy().

{
    for (auto& factory : m_Factories)
    {
        if (factory->GetId() == id)
        {
            return factory.get();
        }
    }

    return nullptr;
}

RegisterFactory()

void RegisterFactory

(

std::unique_ptr< ITensorHandleFactory >

allocator

)

Register a TensorHandleFactory and transfer ownership.

Definition at line 12 of file TensorHandleFactoryRegistry.cpp.

Referenced by ClBackend::CreateWorkloadFactory(), NeonBackend::CreateWorkloadFactory(), RefBackend::CreateWorkloadFactory(), MockImportBackend::CreateWorkloadFactory(), ClBackend::RegisterTensorHandleFactories(), RefBackend::RegisterTensorHandleFactories(), NeonBackend::RegisterTensorHandleFactories(), MockImportBackend::RegisterTensorHandleFactories(), and sdb::SampleDynamicBackendId().

{
    if (!newFactory)
    {
        return;
    }

    ITensorHandleFactory::FactoryId id = newFactory->GetId();

    // Don't register duplicates
    for (auto& registeredFactory : m_Factories)
    {
        if (id == registeredFactory->GetId())
        {
            return;
        }
    }

    // Take ownership of the new allocator
    m_Factories.push_back(std::move(newFactory));
}

RegisterMemoryManager()

void RegisterMemoryManager

(

std::shared_ptr< IMemoryManager >

memoryManger

)

Register a memory manager with shared ownership.

Definition at line 34 of file TensorHandleFactoryRegistry.cpp.

Referenced by ClBackend::CreateWorkloadFactory(), NeonBackend::CreateWorkloadFactory(), RefBackend::CreateWorkloadFactory(), MockImportBackend::CreateWorkloadFactory(), ClBackend::RegisterTensorHandleFactories(), RefBackend::RegisterTensorHandleFactories(), NeonBackend::RegisterTensorHandleFactories(), MockImportBackend::RegisterTensorHandleFactories(), and sdb::SampleDynamicBackendId().

{
    m_MemoryManagers.push_back(memoryManger);
}

ReleaseMemory()

void ReleaseMemory

(

)

Release memory required for inference.

Definition at line 60 of file TensorHandleFactoryRegistry.cpp.

{
    for (auto& mgr : m_MemoryManagers)
    {
        mgr->Release();
    }
}

---

The documentation for this class was generated from the following files:

• src/backends/backendsCommon/TensorHandleFactoryRegistry.hpp
• src/backends/backendsCommon/TensorHandleFactoryRegistry.cpp