xref: /aosp_15_r20/external/armnn/tests/MemoryStrategyBenchmark/TestStrategy.hpp (revision 89c4ff92f2867872bb9e2354d150bf0c8c502810)

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