From: 박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Date: Thu, 8 Aug 2019 04:30:28 +0000 (+0900)
Subject: [moco-tf] Canonicalize ConcatV2 node (#6377)
X-Git-Tag: submit/tizen/20190809.050447~33
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=0c7df712be5787bf6129816a5f87abd47d1d64d0;p=platform%2Fcore%2Fml%2Fnnfw.git

[moco-tf] Canonicalize ConcatV2 node (#6377)

This will enable Canonicalization of TFConcatV2 to canonical TensorConcat

Signed-off-by: SaeHie Park <saehie.park@samsung.com>
---

diff --git a/compiler/moco-tf/src/Canonicalizer.cpp b/compiler/moco-tf/src/Canonicalizer.cpp
index 288b40d..e9d3d16 100644
--- a/compiler/moco-tf/src/Canonicalizer.cpp
+++ b/compiler/moco-tf/src/Canonicalizer.cpp
@@ -23,6 +23,7 @@
 #include "Canonicalization/AddCanonicalizer.h"
 #include "Canonicalization/AvgPoolCanonicalizer.h"
 #include "Canonicalization/BiasAddCanonicalizer.h"
+#include "Canonicalization/ConcatV2Canonicalizer.h"
 #include "Canonicalization/ConstCanonicalizer.h"
 #include "Canonicalization/Conv2DCanonicalizer.h"
 #include "Canonicalization/DepthwiseConv2dNativeCanonicalizer.h"
@@ -77,6 +78,7 @@ void Canonicalizer::canonicalize(loco::Graph *g) const
   phase.emplace_back(stdex::make_unique<AddCanonicalizer>());
   phase.emplace_back(stdex::make_unique<AvgPoolCanonicalizer>());
   phase.emplace_back(stdex::make_unique<BiasAddCanonicalizer>());
+  phase.emplace_back(stdex::make_unique<ConcatV2Canonicalizer>());
   phase.emplace_back(stdex::make_unique<ConstCanonicalizer>());
   phase.emplace_back(stdex::make_unique<Conv2DCanonicalizer>());
   phase.emplace_back(stdex::make_unique<DepthwiseConv2dNativeCanonicalizer>());