1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
//
// Copyright © 2021 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//
#include "../Serializer.hpp"
#include "SerializerTestUtils.hpp"
#include <armnn/Descriptors.hpp>
#include <armnn/INetwork.hpp>
#include <armnn/IRuntime.hpp>
#include <armnnDeserializer/IDeserializer.hpp>
#include <armnn/utility/IgnoreUnused.hpp>
#include <boost/test/unit_test.hpp>
BOOST_AUTO_TEST_SUITE(SerializerTests)
struct ComparisonModel
{
ComparisonModel(const std::string& layerName,
const armnn::TensorInfo& inputInfo,
const armnn::TensorInfo& outputInfo,
armnn::ComparisonDescriptor& descriptor)
: m_network(armnn::INetwork::Create())
{
armnn::IConnectableLayer* const inputLayer0 = m_network->AddInputLayer(0);
armnn::IConnectableLayer* const inputLayer1 = m_network->AddInputLayer(1);
armnn::IConnectableLayer* const equalLayer = m_network->AddComparisonLayer(descriptor, layerName.c_str());
armnn::IConnectableLayer* const outputLayer = m_network->AddOutputLayer(0);
inputLayer0->GetOutputSlot(0).Connect(equalLayer->GetInputSlot(0));
inputLayer1->GetOutputSlot(0).Connect(equalLayer->GetInputSlot(1));
equalLayer->GetOutputSlot(0).Connect(outputLayer->GetInputSlot(0));
inputLayer0->GetOutputSlot(0).SetTensorInfo(inputInfo);
inputLayer1->GetOutputSlot(0).SetTensorInfo(inputInfo);
equalLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
}
armnn::INetworkPtr m_network;
};
class ComparisonLayerVerifier : public LayerVerifierBase
{
public:
ComparisonLayerVerifier(const std::string& layerName,
const std::vector<armnn::TensorInfo>& inputInfos,
const std::vector<armnn::TensorInfo>& outputInfos,
const armnn::ComparisonDescriptor& descriptor)
: LayerVerifierBase(layerName, inputInfos, outputInfos)
, m_Descriptor (descriptor) {}
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
{
armnn::IgnoreUnused(descriptor, constants, id);
switch (layer->GetType())
{
case armnn::LayerType::Input: break;
case armnn::LayerType::Output: break;
case armnn::LayerType::Comparison:
{
VerifyNameAndConnections(layer, name);
const armnn::ComparisonDescriptor& layerDescriptor =
static_cast<const armnn::ComparisonDescriptor&>(descriptor);
BOOST_CHECK(layerDescriptor.m_Operation == m_Descriptor.m_Operation);
break;
}
default:
{
throw armnn::Exception("Unexpected layer type in Comparison test model");
}
}
}
private:
armnn::ComparisonDescriptor m_Descriptor;
};
BOOST_AUTO_TEST_CASE(SerializeEqual)
{
const std::string layerName("equal");
const armnn::TensorShape shape{2, 1, 2, 4};
const armnn::TensorInfo inputInfo = armnn::TensorInfo(shape, armnn::DataType::Float32);
const armnn::TensorInfo outputInfo = armnn::TensorInfo(shape, armnn::DataType::Boolean);
armnn::ComparisonDescriptor descriptor (armnn::ComparisonOperation::Equal);
ComparisonModel model(layerName, inputInfo, outputInfo, descriptor);
armnn::INetworkPtr deserializedNetwork = DeserializeNetwork(SerializeNetwork(*model.m_network));
BOOST_CHECK(deserializedNetwork);
ComparisonLayerVerifier verifier(layerName, { inputInfo, inputInfo }, { outputInfo }, descriptor);
deserializedNetwork->ExecuteStrategy(verifier);
}
BOOST_AUTO_TEST_CASE(SerializeGreater)
{
const std::string layerName("greater");
const armnn::TensorShape shape{2, 1, 2, 4};
const armnn::TensorInfo inputInfo = armnn::TensorInfo(shape, armnn::DataType::Float32);
const armnn::TensorInfo outputInfo = armnn::TensorInfo(shape, armnn::DataType::Boolean);
armnn::ComparisonDescriptor descriptor (armnn::ComparisonOperation::Greater);
ComparisonModel model(layerName, inputInfo, outputInfo, descriptor);
armnn::INetworkPtr deserializedNetwork = DeserializeNetwork(SerializeNetwork(*model.m_network));
BOOST_CHECK(deserializedNetwork);
ComparisonLayerVerifier verifier(layerName, { inputInfo, inputInfo }, { outputInfo }, descriptor);
deserializedNetwork->ExecuteStrategy(verifier);
}
BOOST_AUTO_TEST_SUITE_END()
|
