From 5c7ab05555d3abd23a11db403090b7563ce388e6 Mon Sep 17 00:00:00 2001 From: ruoyan01 Date: Mon, 4 Mar 2019 14:48:02 +0000 Subject: IVGCVSW-2681 Fix layer name in BatchNormalization deserialization Change-Id: Iec43db205d77dc9864abe1fe85104722a6954d57 Signed-off-by: Ruomei Yan --- src/armnnDeserializer/Deserializer.cpp | 2 +- src/armnnSerializer/test/SerializerTests.cpp | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) (limited to 'src') diff --git a/src/armnnDeserializer/Deserializer.cpp b/src/armnnDeserializer/Deserializer.cpp index 405d95ed18..04c629698a 100644 --- a/src/armnnDeserializer/Deserializer.cpp +++ b/src/armnnDeserializer/Deserializer.cpp @@ -877,7 +877,7 @@ void Deserializer::ParseBatchNormalization(GraphPtr graph, unsigned int layerInd CHECK_VALID_SIZE(outputs.size(), 1); auto outputInfo = ToTensorInfo(outputs[0]); - auto layerName = boost::str(boost::format("BatchNormalization:%1%") % layerIndex); + auto layerName = GetLayerName(graph, layerIndex); auto serializerLayer = graph->layers()->Get(layerIndex)->layer_as_BatchNormalizationLayer(); auto serializerDescriptor = serializerLayer->descriptor(); diff --git a/src/armnnSerializer/test/SerializerTests.cpp b/src/armnnSerializer/test/SerializerTests.cpp index 2751aff1d6..62fa0c6bd8 100644 --- a/src/armnnSerializer/test/SerializerTests.cpp +++ b/src/armnnSerializer/test/SerializerTests.cpp @@ -943,7 +943,7 @@ BOOST_AUTO_TEST_CASE(SerializeDeserializeBatchNormalization) const armnn::ConstTensor&, const char* name) override { - BOOST_TEST(name == "BatchNormalization:1"); + BOOST_TEST(name == "BatchNormalizationLayer"); } }; -- cgit v1.2.1