From 5c885373a4a378ddbdfa50a3377c5faa6d166425 Mon Sep 17 00:00:00 2001
From: =?utf8?q?=EB=B0=95=EC=84=B8=ED=9D=AC/On-Device=20Lab=28SR=29/Princip?=
 =?utf8?q?al=20Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?=
 <saehie.park@samsung.com>
Date: Wed, 7 Aug 2019 14:16:27 +0900
Subject: [PATCH] [moco-tf] Enable Canonicalization of Sub and RealDiv (#6316)

This will enable Canonicalizaton of TFSub and TFRealDiv nodes

Signed-off-by: SaeHie Park <saehie.park@samsung.com>
---
 compiler/moco-tf/src/Canonicalizer.cpp | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/compiler/moco-tf/src/Canonicalizer.cpp b/compiler/moco-tf/src/Canonicalizer.cpp
index accd955..288b40d 100644
--- a/compiler/moco-tf/src/Canonicalizer.cpp
+++ b/compiler/moco-tf/src/Canonicalizer.cpp
@@ -29,10 +29,12 @@
 #include "Canonicalization/IdentityCanonicalizer.h"
 #include "Canonicalization/MaxPoolCanonicalizer.h"
 #include "Canonicalization/MulCanonicalizer.h"
+#include "Canonicalization/RealDivCanonicalizer.h"
 #include "Canonicalization/ReluCanonicalizer.h"
 #include "Canonicalization/Relu6Canonicalizer.h"
 #include "Canonicalization/ReshapeCanonicalizer.h"
 #include "Canonicalization/SqueezeCanonicalizer.h"
+#include "Canonicalization/SubCanonicalizer.h"
 
 #include "Dialect/TFDialect.h"
 #include "Dialect/TFNodes.h"
@@ -81,10 +83,12 @@ void Canonicalizer::canonicalize(loco::Graph *g) const
   phase.emplace_back(stdex::make_unique<IdentityCanonicalizer>());
   phase.emplace_back(stdex::make_unique<MaxPoolCanonicalizer>());
   phase.emplace_back(stdex::make_unique<MulCanonicalizer>());
+  phase.emplace_back(stdex::make_unique<RealDivCanonicalizer>());
   phase.emplace_back(stdex::make_unique<ReluCanonicalizer>());
   phase.emplace_back(stdex::make_unique<Relu6Canonicalizer>());
   phase.emplace_back(stdex::make_unique<ReshapeCanonicalizer>());
   phase.emplace_back(stdex::make_unique<SqueezeCanonicalizer>());
+  phase.emplace_back(stdex::make_unique<SubCanonicalizer>());
   /* TRANSFORM DECLARATION END */
 
   moco::tf::PhaseRunner<moco::tf::PhaseStrategy::Saturate> phase_runner{g};
-- 
2.7.4