ArmNN
 24.02
LayerNameAndTypeCheck Class Reference

#include <TestUtils.hpp>

Public Member Functions

 LayerNameAndTypeCheck (armnn::LayerType layerType, const char *name)
 
bool operator() (const armnn::Layer *const layer)
 

Detailed Description

Definition at line 15 of file TestUtils.hpp.

Constructor & Destructor Documentation

◆ LayerNameAndTypeCheck()

LayerNameAndTypeCheck ( armnn::LayerType  layerType,
const char *  name 
)
inline

Definition at line 18 of file TestUtils.hpp.

19  : m_layerType(layerType)
20  , m_name(name)
21  {}

Member Function Documentation

◆ operator()()

bool operator() ( const armnn::Layer *const  layer)
inline

Definition at line 23 of file TestUtils.hpp.

24  {
25  return (layer->GetNameStr() == m_name &&
26  layer->GetType() == m_layerType);
27  }

References Layer::GetNameStr(), and Layer::GetType().


The documentation for this class was generated from the following file:
armnn::Layer::GetNameStr
const std::string & GetNameStr() const
Definition: Layer.hpp:240
armnn::Layer::GetType
LayerType GetType() const override
Returns the armnn::LayerType of this layer.
Definition: Layer.hpp:286