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