diff options
author | ruoyan01 <ruomei.yan@arm.com> | 2019-02-28 15:09:07 +0000 |
---|---|---|
committer | ruoyan01 <ruomei.yan@arm.com> | 2019-03-01 16:39:27 +0000 |
commit | 8e7fa232b4e637cc02f2ca344b2113c63cdc7e5a (patch) | |
tree | 3c200afe3c7cab37b553ba0461aed4410b7cfbb8 /src/armnnSerializer/test | |
parent | dd2ba7ebf78a75aadd8ddd2ae1a4226ffc4ae4d9 (diff) | |
download | armnn-8e7fa232b4e637cc02f2ca344b2113c63cdc7e5a.tar.gz |
IVGCVSW-2681 Serialize/de-serialize the BatchNormalization layer
Change-Id: I418c4465366742262fb6e6c1eeba76c634beaeb5
Signed-off-by: ruoyan01 <ruomei.yan@arm.com>
Diffstat (limited to 'src/armnnSerializer/test')
-rw-r--r-- | src/armnnSerializer/test/SerializerTests.cpp | 68 |
1 files changed, 68 insertions, 0 deletions
diff --git a/src/armnnSerializer/test/SerializerTests.cpp b/src/armnnSerializer/test/SerializerTests.cpp index 5f1745bda5..3ef15831b1 100644 --- a/src/armnnSerializer/test/SerializerTests.cpp +++ b/src/armnnSerializer/test/SerializerTests.cpp @@ -888,6 +888,74 @@ BOOST_AUTO_TEST_CASE(SerializeDeserializeBatchToSpaceNd) {outputTensorInfo.GetShape()}); } +BOOST_AUTO_TEST_CASE(SerializeDeserializeBatchNormalization) +{ + class VerifyBatchNormalizationName : public armnn::LayerVisitorBase<armnn::VisitorNoThrowPolicy> + { + public: + void VisitBatchNormalizationLayer(const armnn::IConnectableLayer*, + const armnn::BatchNormalizationDescriptor&, + const armnn::ConstTensor&, + const armnn::ConstTensor&, + const armnn::ConstTensor&, + const armnn::ConstTensor&, + const char* name) override + { + BOOST_TEST(name == "BatchNormalization:1"); + } + }; + + armnn::TensorInfo inputInfo ({ 1, 3, 3, 1 }, armnn::DataType::Float32); + armnn::TensorInfo outputInfo({ 1, 3, 3, 1 }, armnn::DataType::Float32); + + armnn::TensorInfo meanInfo({1}, armnn::DataType::Float32); + armnn::TensorInfo varianceInfo({1}, armnn::DataType::Float32); + armnn::TensorInfo scaleInfo({1}, armnn::DataType::Float32); + armnn::TensorInfo offsetInfo({1}, armnn::DataType::Float32); + + armnn::BatchNormalizationDescriptor descriptor; + descriptor.m_Eps = 0.0010000000475f; + descriptor.m_DataLayout = armnn::DataLayout::NHWC; + + std::vector<float> meanData({5.0}); + std::vector<float> varianceData({2.0}); + std::vector<float> scaleData({1.0}); + std::vector<float> offsetData({0.0}); + + armnn::ConstTensor mean(meanInfo, meanData); + armnn::ConstTensor variance(varianceInfo, varianceData); + armnn::ConstTensor scale(scaleInfo, scaleData); + armnn::ConstTensor offset(offsetInfo, offsetData); + + armnn::INetworkPtr network = armnn::INetwork::Create(); + armnn::IConnectableLayer* const inputLayer = network->AddInputLayer(0); + armnn::IConnectableLayer* const batchNormalizationLayer = network->AddBatchNormalizationLayer( + descriptor, + mean, + variance, + scale, + offset, + "BatchNormalizationLayer"); + armnn::IConnectableLayer* const outputLayer = network->AddOutputLayer(0); + + inputLayer->GetOutputSlot(0).Connect(batchNormalizationLayer->GetInputSlot(0)); + inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo); + + batchNormalizationLayer->GetOutputSlot(0).Connect(outputLayer->GetInputSlot(0)); + batchNormalizationLayer->GetOutputSlot(0).SetTensorInfo(outputInfo); + + armnn::INetworkPtr deserializedNetwork = DeserializeNetwork(SerializeNetwork(*network)); + BOOST_CHECK(deserializedNetwork); + + VerifyBatchNormalizationName nameChecker; + deserializedNetwork->Accept(nameChecker); + + CheckDeserializedNetworkAgainstOriginal(*network, + *deserializedNetwork, + {inputInfo.GetShape()}, + {outputInfo.GetShape()}); +} + BOOST_AUTO_TEST_CASE(SerializeDivision) { class VerifyDivisionName : public armnn::LayerVisitorBase<armnn::VisitorNoThrowPolicy> |