ArmNN
 21.02
LayerVerifierBaseWithDescriptorAndConstants< Descriptor > Class Template Reference

#include <SerializerTestUtils.hpp>

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

Public Member Functions

 LayerVerifierBaseWithDescriptorAndConstants (const std::string &layerName, const std::vector< armnn::TensorInfo > &inputInfos, const std::vector< armnn::TensorInfo > &outputInfos, const Descriptor &descriptor, const std::vector< armnn::ConstTensor > &constants)
 
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 LayerVerifierBaseWithDescriptor< Descriptor >
 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 ()
 

Additional Inherited Members

- Protected Member Functions inherited from LayerVerifierBaseWithDescriptor< Descriptor >
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 inherited from LayerVerifierBaseWithDescriptor< Descriptor >
Descriptor m_Descriptor
 

Detailed Description

template<typename Descriptor>
class LayerVerifierBaseWithDescriptorAndConstants< Descriptor >

Definition at line 107 of file SerializerTestUtils.hpp.

Constructor & Destructor Documentation

◆ LayerVerifierBaseWithDescriptorAndConstants()

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

Definition at line 110 of file SerializerTestUtils.hpp.

115  : LayerVerifierBaseWithDescriptor<Descriptor>(layerName, inputInfos, outputInfos, descriptor)
116  , m_Constants(constants) {}

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 118 of file SerializerTestUtils.hpp.

References CompareConstTensor(), IConnectableLayer::GetType(), armnn::IgnoreUnused(), armnn::Input, armnn::Output, and LayerVerifierBase::VerifyNameAndConnections().

Referenced by BOOST_AUTO_TEST_CASE().

123  {
125 
126  switch (layer->GetType())
127  {
128  case armnn::LayerType::Input: break;
129  case armnn::LayerType::Output: break;
130  default:
131  {
132  this->VerifyNameAndConnections(layer, name);
133  const Descriptor& internalDescriptor = static_cast<const Descriptor&>(descriptor);
134  this->VerifyDescriptor(internalDescriptor);
135 
136  for(std::size_t i = 0; i < constants.size(); i++)
137  {
138  CompareConstTensor(constants[i], m_Constants[i]);
139  }
140  }
141  }
142  }
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.
void CompareConstTensor(const armnn::ConstTensor &tensor1, const armnn::ConstTensor &tensor2)

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