19 struct ComparisonModel
21 ComparisonModel(
const std::string& layerName,
25 : m_network(
armnn::INetwork::Create())
47 ComparisonLayerVerifier(
const std::string& layerName,
48 const std::vector<armnn::TensorInfo>& inputInfos,
49 const std::vector<armnn::TensorInfo>& outputInfos,
52 , m_Descriptor (descriptor) {}
56 const std::vector<armnn::ConstTensor>& constants,
67 VerifyNameAndConnections(layer, name);
70 CHECK(layerDescriptor.
m_Operation == m_Descriptor.m_Operation);
84 TEST_CASE(
"SerializeEqual")
86 const std::string layerName(
"equal");
94 ComparisonModel model(layerName, inputInfo, outputInfo, descriptor);
97 CHECK(deserializedNetwork);
99 ComparisonLayerVerifier verifier(layerName, { inputInfo, inputInfo }, { outputInfo }, descriptor);
100 deserializedNetwork->ExecuteStrategy(verifier);
103 TEST_CASE(
"SerializeGreater")
105 const std::string layerName(
"greater");
113 ComparisonModel model(layerName, inputInfo, outputInfo, descriptor);
116 CHECK(deserializedNetwork);
118 ComparisonLayerVerifier verifier(layerName, { inputInfo, inputInfo }, { outputInfo }, descriptor);
119 deserializedNetwork->ExecuteStrategy(verifier);
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.
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.
std::unique_ptr< INetwork, void(*)(INetwork *network)> INetworkPtr
virtual int Connect(IInputSlot &destination)=0
std::string SerializeNetwork(const armnn::INetwork &network)