12 #if defined(ARMNNREF_ENABLED) 16 #if defined(ARMCOMPUTENEON_ENABLED) 20 #include <doctest/doctest.h> 27 SampleMemoryOptimizerStrategy() =
default;
31 return std::string(
"SampleMemoryOptimizerStrategy");
39 std::vector<armnn::MemBin>
Optimize(std::vector<armnn::MemBlock>& memBlocks)
41 std::vector<armnn::MemBin> memBins;
42 memBins.reserve(memBlocks.size());
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"}
TEST_SUITE("TestConstTensorLayerVisitor")
MemoryOptimizerStrategiesMapRef GetMemoryOptimizerStrategies()
std::unique_ptr< IRuntime, void(*)(IRuntime *runtime)> IRuntimePtr
BackendRegistry & BackendRegistryInstance()
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.
virtual MemBlockStrategyType GetMemBlockStrategyType() const =0
void DeregisterMemoryOptimizerStrategy(const BackendId &id)
Struct for the users to pass backend specific options.
virtual std::vector< MemBin > Optimize(std::vector< MemBlock > &memBlocks)=0
virtual std::string GetName() const =0