diff options
author | ruoyan01 <ruomei.yan@arm.com> | 2019-03-04 14:48:02 +0000 |
---|---|---|
committer | Ruomei Yan <ruomei.yan@arm.com> | 2019-03-04 15:38:09 +0000 |
commit | 5c7ab05555d3abd23a11db403090b7563ce388e6 (patch) | |
tree | b744c997524f24e2ae721018c7d47f137a4805c0 /src/armnnDeserializer | |
parent | a1ed73a7c0780584a957509ddc4e65ad4de22c39 (diff) | |
download | armnn-5c7ab05555d3abd23a11db403090b7563ce388e6.tar.gz |
IVGCVSW-2681 Fix layer name in BatchNormalization deserialization
Change-Id: Iec43db205d77dc9864abe1fe85104722a6954d57
Signed-off-by: Ruomei Yan <ruomei.yan@arm.com>
Diffstat (limited to 'src/armnnDeserializer')
-rw-r--r-- | src/armnnDeserializer/Deserializer.cpp | 2 |
1 files changed, 1 insertions, 1 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(); |