6 #include "../Serializer.hpp" 15 #include <boost/test/unit_test.hpp> 20 struct ComparisonModel
22 ComparisonModel(
const std::string& layerName,
26 : m_network(
armnn::INetwork::Create())
48 ComparisonLayerVerifier(
const std::string& layerName,
49 const std::vector<armnn::TensorInfo>& inputInfos,
50 const std::vector<armnn::TensorInfo>& outputInfos,
53 , m_Descriptor (descriptor) {}
57 const std::vector<armnn::ConstTensor>& constants,
68 VerifyNameAndConnections(layer, name);
71 BOOST_CHECK(layerDescriptor.
m_Operation == m_Descriptor.m_Operation);
87 const std::string layerName(
"equal");
95 ComparisonModel model(layerName, inputInfo, outputInfo, descriptor);
98 BOOST_CHECK(deserializedNetwork);
100 ComparisonLayerVerifier verifier(layerName, { inputInfo, inputInfo }, { outputInfo }, descriptor);
101 deserializedNetwork->ExecuteStrategy(verifier);
106 const std::string layerName(
"greater");
114 ComparisonModel model(layerName, inputInfo, outputInfo, descriptor);
117 BOOST_CHECK(deserializedNetwork);
119 ComparisonLayerVerifier verifier(layerName, { inputInfo, inputInfo }, { outputInfo }, descriptor);
120 deserializedNetwork->ExecuteStrategy(verifier);
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
Interface for a layer that is connectable to other layers via InputSlots and OutputSlots.
armnn::INetworkPtr DeserializeNetwork(const std::string &serializerString)
A ComparisonDescriptor for the ComparisonLayer.
Copyright (c) 2021 ARM Limited and Contributors.
void IgnoreUnused(Ts &&...)
int LayerBindingId
Type of identifiers for bindable layers (inputs, outputs).
virtual void SetTensorInfo(const TensorInfo &tensorInfo)=0
Base class for all descriptors.
virtual LayerType GetType() const =0
Returns the armnn::LayerType of this layer.
ComparisonOperation m_Operation
Specifies the comparison operation to execute.
BOOST_AUTO_TEST_SUITE_END()
Base class for all ArmNN exceptions so that users can filter to just those.
virtual const IInputSlot & GetInputSlot(unsigned int index) const =0
Get a const input slot handle by slot index.
virtual const IOutputSlot & GetOutputSlot(unsigned int index) const =0
Get the const output slot handle by slot index.
BOOST_AUTO_TEST_CASE(SerializeEqual)
std::unique_ptr< INetwork, void(*)(INetwork *network)> INetworkPtr
virtual int Connect(IInputSlot &destination)=0
std::string SerializeNetwork(const armnn::INetwork &network)