aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/armnnDeserializer/Deserializer.cpp2
-rw-r--r--src/armnnSerializer/test/SerializerTests.cpp2
2 files changed, 2 insertions, 2 deletions
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");
}
};