MemoryOptimizerStrategyFactory.hpp

Go to the documentation of this file.

//
// Copyright © 2021 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//

#pragma once

#include <armnn/backends/IMemoryOptimizerStrategy.hpp>

#include <algorithm>

namespace armnn
{

struct IMemoryOptimizerStrategyFactory
{
    virtual ~IMemoryOptimizerStrategyFactory() = default;
    virtual std::unique_ptr<IMemoryOptimizerStrategy> CreateMemoryOptimizerStrategy() = 0;
};

template <typename T>
struct StrategyFactory : public IMemoryOptimizerStrategyFactory
{
    std::unique_ptr<IMemoryOptimizerStrategy> CreateMemoryOptimizerStrategy() override
    {
        return std::make_unique<T>();
    }
};

} // namespace armnn