SerializerTestUtils.hpp File Reference

#include <armnn/Descriptors.hpp>
#include <armnn/INetwork.hpp>
#include <armnn/TypesUtils.hpp>
#include <armnnDeserializer/IDeserializer.hpp>
#include <armnn/utility/IgnoreUnused.hpp>
#include <random>
#include <vector>
#include <cstdlib>
#include <doctest/doctest.h>

Go to the source code of this file.

Classes
class	LayerVerifierBase
class	LayerVerifierBaseWithDescriptor< Descriptor >
class	LayerVerifierBaseWithDescriptorAndConstants< Descriptor >

Functions
armnn::INetworkPtr	DeserializeNetwork (const std::string &serializerString)
std::string	SerializeNetwork (const armnn::INetwork &network)
void	CompareConstTensor (const armnn::ConstTensor &tensor1, const armnn::ConstTensor &tensor2)
template<typename T >
void	CompareConstTensorData (const void *data1, const void *data2, unsigned int numElements)

Function Documentation

CompareConstTensor()

void CompareConstTensor	(	const armnn::ConstTensor &	tensor1,
		const armnn::ConstTensor &	tensor2
	)

Definition at line 122 of file SerializerTestUtils.cpp.

References armnn::Boolean, armnn::Float32, BaseTensor< MemoryType >::GetDataType(), armnn::GetDataTypeName(), BaseTensor< MemoryType >::GetMemoryArea(), BaseTensor< MemoryType >::GetNumElements(), BaseTensor< MemoryType >::GetShape(), armnn::QAsymmU8, armnn::QSymmS8, and armnn::Signed32.

Referenced by LayerVerifierBaseWithDescriptorAndConstants< Descriptor >::ExecuteStrategy(), and TEST_SUITE().

{
    CHECK(tensor1.GetShape() == tensor2.GetShape());
    CHECK(GetDataTypeName(tensor1.GetDataType()) == GetDataTypeName(tensor2.GetDataType()));

    switch (tensor1.GetDataType())
    {
        case armnn::DataType::Float32:
            CompareConstTensorData<const float*>(
                tensor1.GetMemoryArea(), tensor2.GetMemoryArea(), tensor1.GetNumElements());
            break;
        case armnn::DataType::QAsymmU8:
        case armnn::DataType::Boolean:
            CompareConstTensorData<const uint8_t*>(
                tensor1.GetMemoryArea(), tensor2.GetMemoryArea(), tensor1.GetNumElements());
            break;
        case armnn::DataType::QSymmS8:
            CompareConstTensorData<const int8_t*>(
                tensor1.GetMemoryArea(), tensor2.GetMemoryArea(), tensor1.GetNumElements());
            break;
        case armnn::DataType::Signed32:
            CompareConstTensorData<const int32_t*>(
                tensor1.GetMemoryArea(), tensor2.GetMemoryArea(), tensor1.GetNumElements());
            break;
        default:
            // Note that Float16 is not yet implemented
            MESSAGE("Unexpected datatype");
            CHECK(false);
    }
}

CompareConstTensorData()

void CompareConstTensorData	(	const void *	data1,
		const void *	data2,
		unsigned int	numElements
	)

Definition at line 93 of file SerializerTestUtils.hpp.

{
    T typedData1 = static_cast<T>(data1);
    T typedData2 = static_cast<T>(data2);
    CHECK(typedData1);
    CHECK(typedData2);

    for (unsigned int i = 0; i < numElements; i++)
    {
        CHECK(typedData1[i] == typedData2[i]);
    }
}

DeserializeNetwork()

armnn::INetworkPtr DeserializeNetwork

(

const std::string &

serializerString

)

Definition at line 153 of file SerializerTestUtils.cpp.

Referenced by TEST_SUITE().

{
    std::vector<std::uint8_t> const serializerVector{serializerString.begin(), serializerString.end()};
    return IDeserializer::Create()->CreateNetworkFromBinary(serializerVector);
}

SerializeNetwork()

std::string SerializeNetwork

(

const armnn::INetwork &

network

)

Definition at line 159 of file SerializerTestUtils.cpp.

References ISerializer::Create().

Referenced by TEST_SUITE().

{
    armnnSerializer::ISerializerPtr serializer = armnnSerializer::ISerializer::Create();

    serializer->Serialize(network);

    std::stringstream stream;
    serializer->SaveSerializedToStream(stream);

    std::string serializerString{stream.str()};
    return serializerString;
}