commit e242f2dc646f41e9162aaaf74e057ce39fcb92df
author Jim Flynn <jim.flynn@arm.com> Wed May 22 14:24:13 2019 +0100
committer Jim Flynn <jim.flynn@arm.com> Tue May 28 17:50:33 2019 +0100
tree d6c49b559c34d1d306b1e901501dded1c18f71c5
parent 2f2778f36e59537bbd47fb8b21e73c6c5a949584

IVGCVSW-3119 Rename MergerLayer to ConcatLayer

!android-nn-driver:1210

Change-Id: I940b3b9e421c92bfd55ae996f7bc54ac077f2604
Signed-off-by: Jim Flynn <jim.flynn@arm.com>

src/armnnSerializer/Serializer.cpp

diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp
index 865ed7a..c49f6f9 100644
--- a/src/armnnSerializer/Serializer.cpp
+++ b/src/armnnSerializer/Serializer.cpp
@@ -515,17 +515,24 @@
 }
 
 void SerializerVisitor::VisitMergerLayer(const armnn::IConnectableLayer* layer,
-                                         const armnn::OriginsDescriptor& mergerDescriptor,
+                                         const armnn::MergerDescriptor& mergerDescriptor,
                                          const char* name)
 {
-    auto flatBufferMergerBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Merger);
+    VisitConcatLayer(layer, mergerDescriptor, name);
+}
+
+void SerializerVisitor::VisitConcatLayer(const armnn::IConnectableLayer* layer,
+                                         const armnn::ConcatDescriptor& concatDescriptor,
+                                         const char* name)
+{
+    auto flatBufferConcatBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Concat);
 
     std::vector<flatbuffers::Offset<UintVector>> views;
-    for (unsigned int v = 0; v < mergerDescriptor.GetNumViews(); ++v)
+    for (unsigned int v = 0; v < concatDescriptor.GetNumViews(); ++v)
     {
-        const uint32_t* origin = mergerDescriptor.GetViewOrigin(v);
+        const uint32_t* origin = concatDescriptor.GetViewOrigin(v);
         std::vector<uint32_t> origins;
-        for (unsigned int d = 0; d < mergerDescriptor.GetNumDimensions(); ++d)
+        for (unsigned int d = 0; d < concatDescriptor.GetNumDimensions(); ++d)
         {
             origins.push_back(origin[d]);
         }
@@ -534,17 +541,17 @@
         views.push_back(uintVector);
     }
 
-    auto flatBufferMergerDescriptor = CreateOriginsDescriptor(m_flatBufferBuilder,
-                                                              mergerDescriptor.GetConcatAxis(),
-                                                              mergerDescriptor.GetNumViews(),
-                                                              mergerDescriptor.GetNumDimensions(),
+    auto flatBufferConcatDescriptor = CreateOriginsDescriptor(m_flatBufferBuilder,
+                                                              concatDescriptor.GetConcatAxis(),
+                                                              concatDescriptor.GetNumViews(),
+                                                              concatDescriptor.GetNumDimensions(),
                                                               m_flatBufferBuilder.CreateVector(views));
 
-    auto flatBufferLayer = CreateMergerLayer(m_flatBufferBuilder,
-                                             flatBufferMergerBaseLayer,
-                                             flatBufferMergerDescriptor);
+    auto flatBufferLayer = CreateConcatLayer(m_flatBufferBuilder,
+                                             flatBufferConcatBaseLayer,
+                                             flatBufferConcatDescriptor);
 
-    CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_MergerLayer);
+    CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_ConcatLayer);
 }
 
 void SerializerVisitor::VisitMultiplicationLayer(const armnn::IConnectableLayer* layer, const char* name)