21.11
|
#include <TestLayerVisitor.hpp>
Public Member Functions | |
TestLayerVisitor (const char *name) | |
![]() | |
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 |
![]() | |
virtual void | FinishStrategy () |
Protected Member Functions | |
virtual | ~TestLayerVisitor () |
void | CheckLayerName (const char *name) |
void | CheckLayerPointer (const IConnectableLayer *layer) |
void | CheckConstTensors (const ConstTensor &expected, const ConstTensor &actual) |
void | CheckConstTensors (const ConstTensor &expected, const ConstTensorHandle &actual) |
void | CheckConstTensorPtrs (const std::string &name, const ConstTensor *expected, const ConstTensor *actual) |
void | CheckConstTensorPtrs (const std::string &name, const ConstTensor *expected, const std::shared_ptr< ConstTensorHandle > actual) |
void | CheckOptionalConstTensors (const Optional< ConstTensor > &expected, const Optional< ConstTensor > &actual) |
![]() | |
virtual | ~StrategyBase () |
![]() | |
IStrategy () | |
virtual | ~IStrategy () |
Additional Inherited Members | |
![]() | |
NoThrowStrategy | m_DefaultStrategy |
Definition at line 14 of file TestLayerVisitor.hpp.
|
inlineprotectedvirtual |
Definition at line 17 of file TestLayerVisitor.hpp.
References TestLayerVisitor::CheckConstTensorPtrs(), TestLayerVisitor::CheckConstTensors(), TestLayerVisitor::CheckLayerName(), TestLayerVisitor::CheckLayerPointer(), and TestLayerVisitor::CheckOptionalConstTensors().
|
inlineexplicit |
Definition at line 41 of file TestLayerVisitor.hpp.
|
protected |
Definition at line 90 of file TestLayerVisitor.cpp.
References TestLayerVisitor::CheckConstTensors().
Referenced by LstmVisitor::CheckInputParameters(), TestQuantizedLstmLayerVisitor::CheckInputParameters(), and TestLayerVisitor::~TestLayerVisitor().
|
protected |
Definition at line 72 of file TestLayerVisitor.cpp.
References TestLayerVisitor::CheckConstTensors().
|
protected |
Definition at line 34 of file TestLayerVisitor.cpp.
References BaseTensor< MemoryType >::GetInfo(), BaseTensor< MemoryType >::GetMemoryArea(), BaseTensor< MemoryType >::GetNumBytes(), BaseTensor< MemoryType >::GetNumDimensions(), and BaseTensor< MemoryType >::GetNumElements().
Referenced by TestLayerVisitor::CheckConstTensorPtrs(), TestLayerVisitor::CheckOptionalConstTensors(), TestConvolution2dLayerVisitor::ExecuteStrategy(), TestDepthwiseConvolution2dLayerVisitor::ExecuteStrategy(), TestBatchNormalizationLayerVisitor::ExecuteStrategy(), TestConstantLayerVisitor::ExecuteStrategy(), and TestLayerVisitor::~TestLayerVisitor().
|
protected |
Definition at line 52 of file TestLayerVisitor.cpp.
References BaseTensor< MemoryType >::GetInfo(), BaseTensor< MemoryType >::GetMemoryArea(), BaseTensor< MemoryType >::GetNumBytes(), BaseTensor< MemoryType >::GetNumDimensions(), BaseTensor< MemoryType >::GetNumElements(), ConstTensorHandle::GetTensorInfo(), ConstTensorHandle::Map(), and ConstTensorHandle::Unmap().
|
protected |
Definition at line 13 of file TestLayerVisitor.cpp.
Referenced by TestInputLayerVisitor::ExecuteStrategy(), TestConvolution2dLayerVisitor::ExecuteStrategy(), TestOutputLayerVisitor::ExecuteStrategy(), TestDepthwiseConvolution2dLayerVisitor::ExecuteStrategy(), TestFullyConnectedLayerVistor::ExecuteStrategy(), TestBatchNormalizationLayerVisitor::ExecuteStrategy(), TestConstantLayerVisitor::ExecuteStrategy(), TestLstmLayerVisitor::ExecuteStrategy(), TestQLstmLayerVisitor::ExecuteStrategy(), TestQuantizedLstmLayerVisitor::ExecuteStrategy(), and TestLayerVisitor::~TestLayerVisitor().
|
protected |
Definition at line 29 of file TestLayerVisitor.cpp.
Referenced by TestInputLayerVisitor::ExecuteStrategy(), TestConvolution2dLayerVisitor::ExecuteStrategy(), TestOutputLayerVisitor::ExecuteStrategy(), TestDepthwiseConvolution2dLayerVisitor::ExecuteStrategy(), TestFullyConnectedLayerVistor::ExecuteStrategy(), TestBatchNormalizationLayerVisitor::ExecuteStrategy(), TestConstantLayerVisitor::ExecuteStrategy(), TestLstmLayerVisitor::ExecuteStrategy(), TestQLstmLayerVisitor::ExecuteStrategy(), TestQuantizedLstmLayerVisitor::ExecuteStrategy(), and TestLayerVisitor::~TestLayerVisitor().
|
protected |
Definition at line 108 of file TestLayerVisitor.cpp.
References TestLayerVisitor::CheckConstTensors(), OptionalBase::has_value(), and OptionalReferenceSwitch< std::is_reference< T >::value, T >::value().
Referenced by TestLayerVisitor::~TestLayerVisitor().