| // |
| // 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 |