diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2019-02-21 09:01:41 +0000 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2019-02-21 09:57:44 +0000 |
commit | dbb0c0ca0c8425886ee3a2095e0ced07099134f9 (patch) | |
tree | 2d85dd36b0c23c2fda9f6898a7ab8a9f74f4520c /src/armnnSerializer/test | |
parent | 5324782821dce525cf7c1636c659f998fae2fb85 (diff) | |
download | armnn-dbb0c0ca0c8425886ee3a2095e0ced07099134f9.tar.gz |
IVGCVSW-2639 Add Serializer & Deserializer for Fully Connected
* Added FullyConnectedLayer to Serializer Schema Schema.fbs
* Added FullyConnected serialization and deserialization support
* Added FullyConnected serialization and deserialization unit tests
Change-Id: I8ef14f9728158f849fa4d1a8d05a1a4170cd5b41
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Diffstat (limited to 'src/armnnSerializer/test')
-rw-r--r-- | src/armnnSerializer/test/SerializerTests.cpp | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/src/armnnSerializer/test/SerializerTests.cpp b/src/armnnSerializer/test/SerializerTests.cpp index 822f9c7e00..ede24baf9e 100644 --- a/src/armnnSerializer/test/SerializerTests.cpp +++ b/src/armnnSerializer/test/SerializerTests.cpp @@ -404,4 +404,45 @@ BOOST_AUTO_TEST_CASE(SerializeDeserializePermute) outputTensorInfo.GetShape()); } +BOOST_AUTO_TEST_CASE(SerializeDeserializeFullyConnected) +{ + armnn::TensorInfo inputInfo ({ 2, 5, 1, 1 }, armnn::DataType::Float32); + armnn::TensorInfo outputInfo({ 2, 3 }, armnn::DataType::Float32); + + armnn::TensorInfo weightsInfo({ 5, 3 }, armnn::DataType::Float32); + armnn::TensorInfo biasesInfo ({ 3 }, armnn::DataType::Float32); + + armnn::FullyConnectedDescriptor descriptor; + descriptor.m_BiasEnabled = true; + descriptor.m_TransposeWeightMatrix = false; + + std::vector<float> weightsData = GenerateRandomData<float>(weightsInfo.GetNumElements()); + std::vector<float> biasesData = GenerateRandomData<float>(biasesInfo.GetNumElements()); + + armnn::ConstTensor weights(weightsInfo, weightsData); + armnn::ConstTensor biases(biasesInfo, biasesData); + + armnn::INetworkPtr network = armnn::INetwork::Create(); + armnn::IConnectableLayer* const inputLayer = network->AddInputLayer(0, "input"); + armnn::IConnectableLayer* const fullyConnectedLayer = network->AddFullyConnectedLayer(descriptor, + weights, + biases, + "fully_connected"); + armnn::IConnectableLayer* const outputLayer = network->AddOutputLayer(0, "output"); + + inputLayer->GetOutputSlot(0).Connect(fullyConnectedLayer->GetInputSlot(0)); + inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo); + + fullyConnectedLayer->GetOutputSlot(0).Connect(outputLayer->GetInputSlot(0)); + fullyConnectedLayer->GetOutputSlot(0).SetTensorInfo(outputInfo); + + armnn::INetworkPtr deserializedNetwork = DeserializeNetwork(SerializeNetwork(*network)); + BOOST_CHECK(deserializedNetwork); + + CheckDeserializedNetworkAgainstOriginal(*network, + *deserializedNetwork, + inputInfo.GetShape(), + outputInfo.GetShape()); +} + BOOST_AUTO_TEST_SUITE_END() |