ArmNN  NotReleased
TestLayerVisitor Class Reference

#include <TestLayerVisitor.hpp>

Inheritance diagram for TestLayerVisitor:
LayerVisitorBase< VisitorNoThrowPolicy > ILayerVisitor TestBatchNormalizationLayerVisitor TestConstantLayerVisitor TestConvolution2dLayerVisitor TestDepthwiseConvolution2dLayerVisitor TestFullyConnectedLayerVistor TestInputLayerVisitor TestLstmLayerVisitor TestOutputLayerVisitor TestQuantizedLstmLayerVisitor TestActivationLayerVisitor TestAdditionLayerVisitor TestArgMinMaxLayerVisitor TestBatchToSpaceNdLayerVisitor TestComparisonLayerVisitor TestConcatLayerVisitor TestDepthToSpaceLayerVisitor TestDequantizeLayerVisitor TestDivisionLayerVisitor TestElementwiseUnaryLayerVisitor TestFloorLayerVisitor TestGatherLayerVisitor TestInstanceNormalizationLayerVisitor TestL2NormalizationLayerVisitor TestLogSoftmaxLayerVisitor TestMaximumLayerVisitor TestMeanLayerVisitor TestMergeLayerVisitor TestMinimumLayerVisitor TestMultiplicationLayerVisitor TestNormalizationLayerVisitor TestPadLayerVisitor TestPermuteLayerVisitor TestPooling2dLayerVisitor TestPreluLayerVisitor TestQuantizeLayerVisitor TestReshapeLayerVisitor TestResizeLayerVisitor TestSliceLayerVisitor TestSoftmaxLayerVisitor TestSpaceToBatchNdLayerVisitor TestSpaceToDepthLayerVisitor TestSplitterLayerVisitor TestStackLayerVisitor TestStandInLayerVisitor TestStridedSliceLayerVisitor TestSubtractionLayerVisitor TestSwitchLayerVisitor

Public Member Functions

 TestLayerVisitor (const char *name)
 
- Public Member Functions inherited from LayerVisitorBase< VisitorNoThrowPolicy >
void VisitAbsLayer (const IConnectableLayer *, const char *) override
 
void VisitActivationLayer (const IConnectableLayer *, const ActivationDescriptor &, const char *) override
 
void VisitAdditionLayer (const IConnectableLayer *, const char *) override
 
void VisitArgMinMaxLayer (const IConnectableLayer *, const ArgMinMaxDescriptor &, const char *) override
 
void VisitBatchNormalizationLayer (const IConnectableLayer *, const BatchNormalizationDescriptor &, const ConstTensor &, const ConstTensor &, const ConstTensor &, const ConstTensor &, const char *) override
 
void VisitBatchToSpaceNdLayer (const IConnectableLayer *, const BatchToSpaceNdDescriptor &, const char *) override
 
void VisitComparisonLayer (const IConnectableLayer *, const ComparisonDescriptor &, const char *) override
 
void VisitConcatLayer (const IConnectableLayer *, const ConcatDescriptor &, const char *) override
 
void VisitConstantLayer (const IConnectableLayer *, const ConstTensor &, const char *) override
 
void VisitConvolution2dLayer (const IConnectableLayer *, const Convolution2dDescriptor &, const ConstTensor &, const Optional< ConstTensor > &, const char *) override
 
void VisitDepthToSpaceLayer (const IConnectableLayer *, const DepthToSpaceDescriptor &, const char *) override
 
void VisitDepthwiseConvolution2dLayer (const IConnectableLayer *, const DepthwiseConvolution2dDescriptor &, const ConstTensor &, const Optional< ConstTensor > &, const char *) override
 
void VisitDequantizeLayer (const IConnectableLayer *, const char *) override
 
void VisitDetectionPostProcessLayer (const IConnectableLayer *, const DetectionPostProcessDescriptor &, const ConstTensor &, const char *) override
 
void VisitDivisionLayer (const IConnectableLayer *, const char *) override
 
void VisitElementwiseUnaryLayer (const IConnectableLayer *, const ElementwiseUnaryDescriptor &, const char *) override
 
void VisitEqualLayer (const IConnectableLayer *, const char *) override
 
void VisitFloorLayer (const IConnectableLayer *, const char *) override
 
void VisitFullyConnectedLayer (const IConnectableLayer *, const FullyConnectedDescriptor &, const ConstTensor &, const Optional< ConstTensor > &, const char *) override
 
void VisitGatherLayer (const IConnectableLayer *, const char *) override
 
void VisitGreaterLayer (const IConnectableLayer *, const char *) override
 
void VisitInputLayer (const IConnectableLayer *, LayerBindingId, const char *) override
 
void VisitInstanceNormalizationLayer (const IConnectableLayer *, const InstanceNormalizationDescriptor &, const char *) override
 
void VisitL2NormalizationLayer (const IConnectableLayer *, const L2NormalizationDescriptor &, const char *) override
 
void VisitLogSoftmaxLayer (const IConnectableLayer *, const LogSoftmaxDescriptor &, const char *) override
 
void VisitLstmLayer (const IConnectableLayer *, const LstmDescriptor &, const LstmInputParams &, const char *) override
 
void VisitMaximumLayer (const IConnectableLayer *, const char *) override
 
void VisitMeanLayer (const IConnectableLayer *, const MeanDescriptor &, const char *) override
 
void VisitMergeLayer (const IConnectableLayer *, const char *) override
 
void VisitMergerLayer (const IConnectableLayer *, const MergerDescriptor &, const char *) override
 
void VisitMinimumLayer (const IConnectableLayer *, const char *) override
 
void VisitMultiplicationLayer (const IConnectableLayer *, const char *) override
 
void VisitNormalizationLayer (const IConnectableLayer *, const NormalizationDescriptor &, const char *) override
 
void VisitOutputLayer (const IConnectableLayer *, LayerBindingId, const char *) override
 
void VisitPadLayer (const IConnectableLayer *, const PadDescriptor &, const char *) override
 
void VisitPermuteLayer (const IConnectableLayer *, const PermuteDescriptor &, const char *) override
 
void VisitPooling2dLayer (const IConnectableLayer *, const Pooling2dDescriptor &, const char *) override
 
void VisitPreluLayer (const IConnectableLayer *, const char *) override
 
void VisitQuantizeLayer (const IConnectableLayer *, const char *) override
 
void VisitQuantizedLstmLayer (const IConnectableLayer *, const QuantizedLstmInputParams &, const char *) override
 
void VisitReshapeLayer (const IConnectableLayer *, const ReshapeDescriptor &, const char *) override
 
void VisitResizeBilinearLayer (const IConnectableLayer *, const ResizeBilinearDescriptor &, const char *) override
 
void VisitResizeLayer (const IConnectableLayer *, const ResizeDescriptor &, const char *) override
 
void VisitRsqrtLayer (const IConnectableLayer *, const char *) override
 
void VisitSliceLayer (const IConnectableLayer *, const SliceDescriptor &, const char *) override
 
void VisitSoftmaxLayer (const IConnectableLayer *, const SoftmaxDescriptor &, const char *) override
 
void VisitSpaceToBatchNdLayer (const IConnectableLayer *, const SpaceToBatchNdDescriptor &, const char *) override
 
void VisitSpaceToDepthLayer (const IConnectableLayer *, const SpaceToDepthDescriptor &, const char *) override
 
void VisitSplitterLayer (const IConnectableLayer *, const ViewsDescriptor &, const char *) override
 
void VisitStackLayer (const IConnectableLayer *, const StackDescriptor &, const char *) override
 
void VisitStandInLayer (const IConnectableLayer *, const StandInDescriptor &, const char *) override
 
void VisitStridedSliceLayer (const IConnectableLayer *, const StridedSliceDescriptor &, const char *) override
 
void VisitSubtractionLayer (const IConnectableLayer *, const char *) override
 
void VisitSwitchLayer (const IConnectableLayer *, const char *) override
 
void VisitTransposeConvolution2dLayer (const IConnectableLayer *, const TransposeConvolution2dDescriptor &, const ConstTensor &, const Optional< ConstTensor > &, const char *) override
 
- Public Member Functions inherited from ILayerVisitor
virtual void StartVisit ()
 
virtual void FinishVisit ()
 

Protected Member Functions

virtual ~TestLayerVisitor ()
 
void CheckLayerName (const char *name)
 
void CheckLayerPointer (const IConnectableLayer *layer)
 
void CheckConstTensors (const ConstTensor &expected, const ConstTensor &actual)
 
void CheckOptionalConstTensors (const Optional< ConstTensor > &expected, const Optional< ConstTensor > &actual)
 
- Protected Member Functions inherited from LayerVisitorBase< VisitorNoThrowPolicy >
 LayerVisitorBase ()
 
virtual ~LayerVisitorBase ()
 
- Protected Member Functions inherited from ILayerVisitor
 ILayerVisitor ()
 
virtual ~ILayerVisitor ()
 

Detailed Description

Definition at line 13 of file TestLayerVisitor.hpp.

Constructor & Destructor Documentation

◆ ~TestLayerVisitor()

◆ TestLayerVisitor()

TestLayerVisitor ( const char *  name)
inlineexplicit

Definition at line 30 of file TestLayerVisitor.hpp.

30  : m_LayerName(name)
31  {
32  if (name == nullptr)
33  {
34  m_LayerName = "";
35  }
36  }

Member Function Documentation

◆ CheckConstTensors()

void CheckConstTensors ( const ConstTensor expected,
const ConstTensor actual 
)
protected

Definition at line 33 of file TestLayerVisitor.cpp.

References BOOST_CHECK(), BaseTensor< MemoryType >::GetInfo(), BaseTensor< MemoryType >::GetMemoryArea(), BaseTensor< MemoryType >::GetNumBytes(), BaseTensor< MemoryType >::GetNumDimensions(), and BaseTensor< MemoryType >::GetNumElements().

Referenced by TestLstmLayerVisitor::CheckConstTensorPtrs(), TestQuantizedLstmLayerVisitor::CheckConstTensorPtrs(), TestLayerVisitor::CheckOptionalConstTensors(), TestBatchNormalizationLayerVisitor::VisitBatchNormalizationLayer(), TestConstantLayerVisitor::VisitConstantLayer(), TestConvolution2dLayerVisitor::VisitConvolution2dLayer(), TestDepthwiseConvolution2dLayerVisitor::VisitDepthwiseConvolution2dLayer(), TestFullyConnectedLayerVistor::VisitFullyConnectedLayer(), and TestLayerVisitor::~TestLayerVisitor().

34 {
35  BOOST_CHECK(expected.GetInfo() == actual.GetInfo());
36  BOOST_CHECK(expected.GetNumDimensions() == actual.GetNumDimensions());
37  BOOST_CHECK(expected.GetNumElements() == actual.GetNumElements());
38  BOOST_CHECK(expected.GetNumBytes() == actual.GetNumBytes());
39  if (expected.GetNumBytes() == actual.GetNumBytes())
40  {
41  //check data is the same byte by byte
42  const unsigned char* expectedPtr = static_cast<const unsigned char*>(expected.GetMemoryArea());
43  const unsigned char* actualPtr = static_cast<const unsigned char*>(actual.GetMemoryArea());
44  for (unsigned int i = 0; i < expected.GetNumBytes(); i++)
45  {
46  BOOST_CHECK(*(expectedPtr + i) == *(actualPtr + i));
47  }
48  }
49 }
BOOST_CHECK(profilingService.GetCurrentState()==ProfilingState::WaitingForAck)

◆ CheckLayerName()

void CheckLayerName ( const char *  name)
protected

◆ CheckLayerPointer()

◆ CheckOptionalConstTensors()

void CheckOptionalConstTensors ( const Optional< ConstTensor > &  expected,
const Optional< ConstTensor > &  actual 
)
protected

Definition at line 51 of file TestLayerVisitor.cpp.

References BOOST_CHECK(), TestLayerVisitor::CheckConstTensors(), OptionalBase::has_value(), and OptionalReferenceSwitch< std::is_reference< T >::value, T >::value().

Referenced by TestConvolution2dLayerVisitor::VisitConvolution2dLayer(), TestDepthwiseConvolution2dLayerVisitor::VisitDepthwiseConvolution2dLayer(), TestFullyConnectedLayerVistor::VisitFullyConnectedLayer(), and TestLayerVisitor::~TestLayerVisitor().

53 {
54  BOOST_CHECK(expected.has_value() == actual.has_value());
55  if (expected.has_value() && actual.has_value())
56  {
57  CheckConstTensors(expected.value(), actual.value());
58  }
59 }
BOOST_CHECK(profilingService.GetCurrentState()==ProfilingState::WaitingForAck)
void CheckConstTensors(const ConstTensor &expected, const ConstTensor &actual)

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