51 #if defined(ARMNNREF_ENABLED) 52 TEST_CASE(
"RefCustomMemoryOptimizerStrategyTest")
54 using namespace armnn;
58 auto customMemoryOptimizerStrategy = std::make_shared<SampleMemoryOptimizerStrategy>();
66 CHECK(optimizerStrategy->GetName() == std::string(
"SampleMemoryOptimizerStrategy"));
74 TEST_CASE(
"CpuRefSetMemoryOptimizerStrategyTest")
76 using namespace armnn;
83 {
"MemoryOptimizerStrategy",
"ConstantMemoryStrategy"}
94 CHECK(optimizerStrategy->GetName() == std::string(
"ConstantMemoryStrategy"));
101 #if defined(ARMCOMPUTENEON_ENABLED) 103 TEST_CASE(
"CpuAccSetMemoryOptimizerStrategyTest")
105 using namespace armnn;
112 {
"MemoryOptimizerStrategy",
"NotExistMemoryOptimizerStrategy"}
static IRuntimePtr Create(const CreationOptions &options)
MemoryOptimizerStrategiesMapRef GetMemoryOptimizerStrategies()
std::unique_ptr< IRuntime, void(*)(IRuntime *runtime)> IRuntimePtr
BackendRegistry & BackendRegistryInstance()
static const BackendId & GetIdStatic()
std::map< BackendId, std::shared_ptr< IMemoryOptimizerStrategy > > m_MemoryOptimizerStrategyMap
A map to define a custom memory optimizer strategy for specific backend Ids.
Copyright (c) 2021 ARM Limited and Contributors.
std::vector< BackendOptions > m_BackendOptions
Pass backend specific options.
void DeregisterMemoryOptimizerStrategy(const BackendId &id)
Struct for the users to pass backend specific options.
static const BackendId & GetIdStatic()