ArmNN
 21.02
LayerVerifierBaseWithDescriptor< Descriptor > Class Template Reference

#include <SerializerTestUtils.hpp>

Inheritance diagram for LayerVerifierBaseWithDescriptor< Descriptor >:
LayerVerifierBase IStrategy LayerVerifierBaseWithDescriptorAndConstants< Descriptor >

Public Member Functions

 LayerVerifierBaseWithDescriptor (const std::string &layerName, const std::vector< armnn::TensorInfo > &inputInfos, const std::vector< armnn::TensorInfo > &outputInfos, const Descriptor &descriptor)
 
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
 
- Public Member Functions inherited from LayerVerifierBase
 LayerVerifierBase (const std::string &layerName, const std::vector< armnn::TensorInfo > &inputInfos, const std::vector< armnn::TensorInfo > &outputInfos)
 
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
 
- Public Member Functions inherited from IStrategy
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)=0
 
virtual void FinishStrategy ()
 

Protected Member Functions

void VerifyDescriptor (const Descriptor &descriptor)
 
- Protected Member Functions inherited from LayerVerifierBase
void VerifyNameAndConnections (const armnn::IConnectableLayer *layer, const char *name)
 
void VerifyConstTensors (const std::string &tensorName, const armnn::ConstTensor *expectedPtr, const armnn::ConstTensor *actualPtr)
 
- Protected Member Functions inherited from IStrategy
 IStrategy ()
 
virtual ~IStrategy ()
 

Protected Attributes

Descriptor m_Descriptor
 

Detailed Description

template<typename Descriptor>
class LayerVerifierBaseWithDescriptor< Descriptor >

Definition at line 51 of file SerializerTestUtils.hpp.

Constructor & Destructor Documentation

◆ LayerVerifierBaseWithDescriptor()

LayerVerifierBaseWithDescriptor ( const std::string &  layerName,
const std::vector< armnn::TensorInfo > &  inputInfos,
const std::vector< armnn::TensorInfo > &  outputInfos,
const Descriptor &  descriptor 
)
inline

Definition at line 54 of file SerializerTestUtils.hpp.

58  : LayerVerifierBase(layerName, inputInfos, outputInfos)
59  , m_Descriptor(descriptor) {}
LayerVerifierBase(const std::string &layerName, const std::vector< armnn::TensorInfo > &inputInfos, const std::vector< armnn::TensorInfo > &outputInfos)

Member Function Documentation

◆ ExecuteStrategy()

void ExecuteStrategy ( const armnn::IConnectableLayer layer,
const armnn::BaseDescriptor descriptor,
const std::vector< armnn::ConstTensor > &  constants,
const char *  name,
const armnn::LayerBindingId  id = 0 
)
inlineoverride

Definition at line 61 of file SerializerTestUtils.hpp.

Referenced by BOOST_AUTO_TEST_CASE(), and ConstantVector2LstmInputParams().

66  {
67  armnn::IgnoreUnused(constants, id);
68  switch (layer->GetType())
69  {
70  case armnn::LayerType::Input: break;
71  case armnn::LayerType::Output: break;
72  default:
73  {
74  VerifyNameAndConnections(layer, name);
75  const Descriptor& internalDescriptor = static_cast<const Descriptor&>(descriptor);
76  VerifyDescriptor(internalDescriptor);
77  break;
78  }
79  }
80  }
void VerifyDescriptor(const Descriptor &descriptor)
void IgnoreUnused(Ts &&...)
void VerifyNameAndConnections(const armnn::IConnectableLayer *layer, const char *name)
virtual LayerType GetType() const =0
Returns the armnn::LayerType of this layer.

◆ VerifyDescriptor()

void VerifyDescriptor ( const Descriptor &  descriptor)
inlineprotected

Definition at line 83 of file SerializerTestUtils.hpp.

Referenced by ConstantVector2LstmInputParams().

84  {
85  BOOST_CHECK(descriptor == m_Descriptor);
86  }

Member Data Documentation

◆ m_Descriptor

Descriptor m_Descriptor
protected

Definition at line 88 of file SerializerTestUtils.hpp.


The documentation for this class was generated from the following file: