BackendRegistry Class Reference

#include <BackendRegistry.hpp>

Inheritance diagram for BackendRegistry:

Classes
struct	StaticRegistryInitializer

Public Types
using	PointerType = IBackendInternalUniquePtr
using	FactoryFunction = std::function< PointerType()>

Public Member Functions
void	Register (const BackendId &id, FactoryFunction factory)
bool	IsBackendRegistered (const BackendId &id) const
FactoryFunction	GetFactory (const BackendId &id) const
size_t	Size () const
BackendIdSet	GetBackendIds () const
std::string	GetBackendIdsAsString () const
void	SetProfilingService (armnn::Optional< profiling::ProfilingService &> profilingService)
	BackendRegistry ()
virtual	~BackendRegistry ()
void	Deregister (const BackendId &id)

Protected Types
using	FactoryStorage = std::unordered_map< BackendId, FactoryFunction >

Static Protected Member Functions
static void	Swap (BackendRegistry &instance, FactoryStorage &other)
	For testing only. More...

Detailed Description

Definition at line 25 of file BackendRegistry.hpp.

Member Typedef Documentation

FactoryFunction

using FactoryFunction = std::function<PointerType()>

Definition at line 29 of file BackendRegistry.hpp.

FactoryStorage

using FactoryStorage = std::unordered_map<BackendId, FactoryFunction>

protected

Definition at line 55 of file BackendRegistry.hpp.

PointerType

using PointerType = IBackendInternalUniquePtr

Definition at line 28 of file BackendRegistry.hpp.

Constructor & Destructor Documentation

BackendRegistry()

BackendRegistry ( )

inline

Definition at line 39 of file BackendRegistry.hpp.

{}

~BackendRegistry()

virtual ~BackendRegistry ( )

inlinevirtual

Definition at line 40 of file BackendRegistry.hpp.

{}

Member Function Documentation

Deregister()

void Deregister

(

const BackendId &

id

)

Definition at line 39 of file BackendRegistry.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), DynamicBackendUtils::DeregisterDynamicBackends(), and MockBackendInitialiser::~MockBackendInitialiser().

{
    m_Factories.erase(id);

    if (m_ProfilingService.has_value() && m_ProfilingService.value().IsProfilingEnabled())
    {
        m_ProfilingService.value().IncrementCounterValue(armnn::profiling::UNREGISTERED_BACKENDS);
    }
}

GetBackendIds()

BackendIdSet GetBackendIds

(

)

const

Definition at line 72 of file BackendRegistry.cpp.

Referenced by BackendRegistry::GetBackendIdsAsString().

{
    BackendIdSet result;
    for (const auto& it : m_Factories)
    {
        result.insert(it.first);
    }
    return result;
}

GetBackendIdsAsString()

std::string GetBackendIdsAsString

(

)

const

Definition at line 82 of file BackendRegistry.cpp.

References BackendRegistry::GetBackendIds().

Referenced by InferenceModel< IParser, TDataType >::AddCommandLineOptions(), main(), and RunCsvTest().

{
    static const std::string delimitator = ", ";

    std::stringstream output;
    for (auto& backendId : GetBackendIds())
    {
        if (output.tellp() != std::streampos(0))
        {
            output << delimitator;
        }
        output << backendId;
    }

    return output.str();
}

GetFactory()

BackendRegistry::FactoryFunction GetFactory

(

const BackendId &

id

)

const

Definition at line 54 of file BackendRegistry.cpp.

References CHECK_LOCATION.

Referenced by BOOST_AUTO_TEST_CASE(), armnn::GetILayerSupportByBackendId(), LoadedNetwork::MakeLoadedNetwork(), armnn::Optimize(), and Runtime::Runtime().

{
    auto it = m_Factories.find(id);
    if (it == m_Factories.end())
    {
        throw InvalidArgumentException(
            std::string(id) + " has no IBackend factory registered",
            CHECK_LOCATION());
    }

    return it->second;
}

IsBackendRegistered()

bool IsBackendRegistered

(

const BackendId &

id

)

const

Definition at line 49 of file BackendRegistry.cpp.

Referenced by armnn::GetILayerSupportByBackendId(), and DynamicBackendUtils::RegisterDynamicBackendsImpl().

{
    return (m_Factories.find(id) != m_Factories.end());
}

Register()

void Register	(	const BackendId &	id,
		BackendRegistry::FactoryFunction	factory
	)

Definition at line 19 of file BackendRegistry.cpp.

References CHECK_LOCATION.

Referenced by BOOST_AUTO_TEST_CASE(), MockBackendInitialiser::MockBackendInitialiser(), DynamicBackendUtils::RegisterDynamicBackendsImpl(), and BackendRegistry::StaticRegistryInitializer::StaticRegistryInitializer().

{
    if (m_Factories.find(id) != m_Factories.end())
    {
        throw InvalidArgumentException(
            std::string(id) + " already registered as IBackend factory",
            CHECK_LOCATION());
    }
    m_Factories[id] = factory;

    if (m_ProfilingService.has_value())
    {
        if (m_ProfilingService.has_value() && m_ProfilingService.value().IsProfilingEnabled())
        {
            m_ProfilingService.value().IncrementCounterValue(armnn::profiling::REGISTERED_BACKENDS);
        }
    }

}

SetProfilingService()

void SetProfilingService

(

armnn::Optional< profiling::ProfilingService &>

profilingService

)

Definition at line 104 of file BackendRegistry.cpp.

References profilingService.

Referenced by Runtime::Runtime(), and Runtime::~Runtime().

{
    m_ProfilingService = profilingService;
}

Size()

size_t Size

(

)

const

Definition at line 67 of file BackendRegistry.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), RegisterSingleDynamicBackendTestImpl(), RuntimeEmptyTestImpl(), and RuntimeInvalidOverridePathTestImpl().

{
    return m_Factories.size();
}

Swap()

void Swap ( BackendRegistry &  instance, BackendRegistry::FactoryStorage &  other  )

staticprotected

For testing only.

Definition at line 99 of file BackendRegistry.cpp.

References armnn::swap().

{
    std::swap(instance.m_Factories, other);
}

---

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

• include/armnn/BackendRegistry.hpp
• src/armnn/BackendRegistry.cpp