StrategyBase.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/INetwork.hpp>
#include <armnn/IStrategy.hpp>
#include <armnn/utility/IgnoreUnused.hpp>
 
namespace armnn
{
 
struct ThrowingStrategy
{
    void Apply(const std::string& errorMessage = "") { throw UnimplementedException(errorMessage); };
};
 
struct NoThrowStrategy
{
    void Apply(const std::string&) {};
};
 
/// Strategy base class with empty implementations.
template <typename DefaultStrategy>
class StrategyBase : public IStrategy
{
protected:
    virtual ~StrategyBase() {};
 
public:
    virtual void ExecuteStrategy(const armnn::IConnectableLayer* layer,
                                 const armnn::BaseDescriptor& descriptor,
                                 const std::vector<armnn::ConstTensor>& constants,
                                 const char* name,
                                 const armnn::LayerBindingId id = 0) override
    {
        armnn::IgnoreUnused(descriptor, constants, id, name);
        switch (layer->GetType())
        {
            default:
            {
                m_DefaultStrategy.Apply(GetLayerTypeAsCString(layer->GetType()));
            }
        }
    }
 
protected:
    DefaultStrategy m_DefaultStrategy;
 
};
 
 
} // namespace armnn