SerializerTestUtils.cpp File Reference

#include "SerializerTestUtils.hpp"
#include "../Serializer.hpp"
#include <doctest/doctest.h>

Go to the source code of this file.

Functions
void	CompareConstTensor (const armnn::ConstTensor &tensor1, const armnn::ConstTensor &tensor2)
armnn::INetworkPtr	DeserializeNetwork (const std::string &serializerString)
std::string	SerializeNetwork (const armnn::INetwork &network)

Function Documentation

CompareConstTensor()

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

Definition at line 116 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);
    }
}

DeserializeNetwork()

armnn::INetworkPtr DeserializeNetwork

(

const std::string &

serializerString

)

Definition at line 147 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 153 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;
}