From 9a8e02dd07667f6bcb2690491f197bc989056d32 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?= Date: Wed, 21 Aug 2019 07:08:38 +0900 Subject: [PATCH] [moco-tf] Enable Canonicalizer for TFStopGradient (#6737) This will add enable TFStopGradient in Canonicalizer Signed-off-by: SaeHie Park --- compiler/moco-tf/src/Canonicalizer.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/compiler/moco-tf/src/Canonicalizer.cpp b/compiler/moco-tf/src/Canonicalizer.cpp index d27aa2a..8721f2d 100644 --- a/compiler/moco-tf/src/Canonicalizer.cpp +++ b/compiler/moco-tf/src/Canonicalizer.cpp @@ -41,6 +41,7 @@ #include "Canonicalization/SqrtCanonicalizer.h" #include "Canonicalization/SquaredDifferenceCanonicalizer.h" #include "Canonicalization/SqueezeCanonicalizer.h" +#include "Canonicalization/StopGradientCanonicalizer.h" #include "Canonicalization/SubCanonicalizer.h" #include "Dialect/TFDialect.h" @@ -103,6 +104,7 @@ void Canonicalizer::canonicalize(loco::Graph *g) const phase.emplace_back(stdex::make_unique()); phase.emplace_back(stdex::make_unique()); phase.emplace_back(stdex::make_unique()); + phase.emplace_back(stdex::make_unique()); phase.emplace_back(stdex::make_unique()); /* TRANSFORM DECLARATION END */ -- 2.7.4