20.11
|
#include <BackendRegistry.hpp>
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... | |
Definition at line 25 of file BackendRegistry.hpp.
using FactoryFunction = std::function<PointerType()> |
Definition at line 29 of file BackendRegistry.hpp.
|
protected |
Definition at line 55 of file BackendRegistry.hpp.
using PointerType = IBackendInternalUniquePtr |
Definition at line 28 of file BackendRegistry.hpp.
|
inline |
Definition at line 39 of file BackendRegistry.hpp.
|
inlinevirtual |
Definition at line 40 of file BackendRegistry.hpp.
void Deregister | ( | const BackendId & | id | ) |
Definition at line 39 of file BackendRegistry.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), DynamicBackendUtils::DeregisterDynamicBackends(), MockBackendInitialiser::~MockBackendInitialiser(), and MockImportBackendInitialiser::~MockImportBackendInitialiser().
BackendIdSet GetBackendIds | ( | ) | const |
Definition at line 72 of file BackendRegistry.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), and BackendRegistry::GetBackendIdsAsString().
std::string GetBackendIdsAsString | ( | ) | const |
Definition at line 82 of file BackendRegistry.cpp.
References BackendRegistry::GetBackendIds().
Referenced by InferenceModel< IParser, TDataType >::AddCommandLineOptions(), main(), and ProgramOptions::ProgramOptions().
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().
bool IsBackendRegistered | ( | const BackendId & | id | ) | const |
Definition at line 49 of file BackendRegistry.cpp.
Referenced by armnn::GetILayerSupportByBackendId(), and DynamicBackendUtils::RegisterDynamicBackendsImpl().
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(), MockImportBackendInitialiser::MockImportBackendInitialiser(), DynamicBackendUtils::RegisterDynamicBackendsImpl(), and BackendRegistry::StaticRegistryInitializer::StaticRegistryInitializer().
void SetProfilingService | ( | armnn::Optional< profiling::ProfilingService &> | profilingService | ) |
Definition at line 104 of file BackendRegistry.cpp.
Referenced by Runtime::Runtime(), and Runtime::~Runtime().
size_t Size | ( | ) | const |
Definition at line 67 of file BackendRegistry.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), RuntimeEmptyTestImpl(), and RuntimeInvalidOverridePathTestImpl().
|
staticprotected |
For testing only.
Definition at line 99 of file BackendRegistry.cpp.
References armnn::swap().