[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 (file)
--- 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>());