TestLayerVisitor Class Reference

#include <TestLayerVisitor.hpp>

Inheritance diagram for TestLayerVisitor:

Public Member Functions
	TestLayerVisitor (const char *name)
Public Member Functions inherited from StrategyBase< NoThrowStrategy >
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
Public Member Functions inherited from IStrategy
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)
Protected Member Functions inherited from StrategyBase< NoThrowStrategy >
virtual	~StrategyBase ()
Protected Member Functions inherited from IStrategy
	IStrategy ()
virtual	~IStrategy ()

Additional Inherited Members
Protected Attributes inherited from StrategyBase< NoThrowStrategy >
NoThrowStrategy	m_DefaultStrategy

Detailed Description

Definition at line 14 of file TestLayerVisitor.hpp.

Constructor & Destructor Documentation

~TestLayerVisitor()

virtual ~TestLayerVisitor ( )

inlineprotectedvirtual

Definition at line 17 of file TestLayerVisitor.hpp.

References TestLayerVisitor::CheckConstTensorPtrs(), TestLayerVisitor::CheckConstTensors(), TestLayerVisitor::CheckLayerName(), TestLayerVisitor::CheckLayerPointer(), and TestLayerVisitor::CheckOptionalConstTensors().

{}

TestLayerVisitor()

TestLayerVisitor ( const char *  name )

inlineexplicit

Definition at line 41 of file TestLayerVisitor.hpp.

                                                : m_LayerName(name)
    {
        if (name == nullptr)
        {
            m_LayerName = "";
        }
    }

Member Function Documentation

CheckConstTensorPtrs() [1/2]

void CheckConstTensorPtrs ( const std::string &  name, const ConstTensor *  expected, const ConstTensor *  actual  )

protected

Definition at line 90 of file TestLayerVisitor.cpp.

References TestLayerVisitor::CheckConstTensors().

Referenced by LstmVisitor::CheckInputParameters(), TestQuantizedLstmLayerVisitor::CheckInputParameters(), and TestLayerVisitor::~TestLayerVisitor().

{
    if (expected == nullptr)
    {
        CHECK_MESSAGE(actual == nullptr, (name + " actual should have been a nullptr"));
    }
    else
    {
        CHECK_MESSAGE(actual != nullptr, (name + " actual should have been set"));
        if (actual != nullptr)
        {
            CheckConstTensors(*expected, *actual);
        }
    }
}

CheckConstTensorPtrs() [2/2]

void CheckConstTensorPtrs ( const std::string &  name, const ConstTensor *  expected, const std::shared_ptr< ConstTensorHandle >  actual  )

protected

Definition at line 72 of file TestLayerVisitor.cpp.

References TestLayerVisitor::CheckConstTensors().

{
    if (expected == nullptr)
    {
        CHECK_MESSAGE(actual == nullptr, (name + " actual should have been a nullptr"));
    }
    else
    {
        CHECK_MESSAGE(actual != nullptr, (name + " actual should have been set"));
        if (actual != nullptr)
        {
            CheckConstTensors(*expected, *actual);
        }
    }
}

CheckConstTensors() [1/2]

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

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(), TestBatchNormalizationLayerVisitor::ExecuteStrategy(), TestConstantLayerVisitor::ExecuteStrategy(), and TestLayerVisitor::~TestLayerVisitor().

{
    CHECK(expected.GetInfo() == actual.GetInfo());
    CHECK(expected.GetNumDimensions() == actual.GetNumDimensions());
    CHECK(expected.GetNumElements() == actual.GetNumElements());
    CHECK(expected.GetNumBytes() == actual.GetNumBytes());
    if (expected.GetNumBytes() == actual.GetNumBytes())
    {
        //check data is the same byte by byte
        const unsigned char* expectedPtr = static_cast<const unsigned char*>(expected.GetMemoryArea());
        const unsigned char* actualPtr = static_cast<const unsigned char*>(actual.GetMemoryArea());
        for (unsigned int i = 0; i < expected.GetNumBytes(); i++)
        {
            CHECK(*(expectedPtr + i) == *(actualPtr + i));
        }
    }
}

CheckConstTensors() [2/2]

void CheckConstTensors ( const ConstTensor &  expected, const ConstTensorHandle &  actual  )

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().

{
    auto& actualInfo = actual.GetTensorInfo();
    CHECK(expected.GetInfo() == actualInfo);
    CHECK(expected.GetNumDimensions() == actualInfo.GetNumDimensions());
    CHECK(expected.GetNumElements() == actualInfo.GetNumElements());
    CHECK(expected.GetNumBytes() == actualInfo.GetNumBytes());
    if (expected.GetNumBytes() == actualInfo.GetNumBytes())
    {
        //check data is the same byte by byte
        const unsigned char* expectedPtr = static_cast<const unsigned char*>(expected.GetMemoryArea());
        const unsigned char* actualPtr = static_cast<const unsigned char*>(actual.Map(true));
        for (unsigned int i = 0; i < expected.GetNumBytes(); i++)
        {
            CHECK(*(expectedPtr + i) == *(actualPtr + i));
        }
        actual.Unmap();
    }
}

CheckLayerName()

void CheckLayerName ( const char *  name )

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().

{
    if (name == nullptr)
    {
        CHECK(m_LayerName == nullptr);
    }
    else if (m_LayerName == nullptr)
    {
        CHECK(name == nullptr);
    }
    else
    {
        CHECK_EQ(std::string(m_LayerName), std::string(name));
    }
}

CheckLayerPointer()

void CheckLayerPointer ( const IConnectableLayer *  layer )

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().

{
    CHECK(layer != nullptr);
}

CheckOptionalConstTensors()

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

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().

{
    CHECK(expected.has_value() == actual.has_value());
    if (expected.has_value() && actual.has_value())
    {
        CheckConstTensors(expected.value(), actual.value());
    }
}

---

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

• src/armnn/test/TestLayerVisitor.hpp
• src/armnn/test/TestLayerVisitor.cpp