hmz007
36ed224bac
Signed-off-by: hmz007 <hmz007@gmail.com> Change-Id: Ib87fdbe7a0be7dc961af3500fe9ea4589a127f9f
31 lines
881 B
C++
31 lines
881 B
C++
//
|
|
// Copyright © 2021 Arm Ltd and Contributors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
//
|
|
#pragma once
|
|
|
|
#include <armnn/Types.hpp>
|
|
#include <armnn/backends/IMemoryOptimizerStrategy.hpp>
|
|
|
|
namespace armnn
|
|
{
|
|
// ConstLayerMemoryOptimizer: Create a unique MemBin for each MemBlock and assign it an offset of 0
|
|
class TestStrategy : public IMemoryOptimizerStrategy
|
|
{
|
|
public:
|
|
TestStrategy()
|
|
: m_Name(std::string("TestStrategy"))
|
|
, m_MemBlockStrategyType(MemBlockStrategyType::MultiAxisPacking) {}
|
|
|
|
std::string GetName() const override;
|
|
|
|
MemBlockStrategyType GetMemBlockStrategyType() const override;
|
|
|
|
std::vector<MemBin> Optimize(std::vector<MemBlock>& memBlocks) override;
|
|
|
|
private:
|
|
std::string m_Name;
|
|
MemBlockStrategyType m_MemBlockStrategyType;
|
|
};
|
|
|
|
} // namespace armnn
|