From 1e6adff6915045f231185df8d77a79169abd4407 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: Mon, 5 Aug 2019 16:52:36 +0900 Subject: [PATCH] [moco-tf] Canonicalize TFRelu and TFRelu6 (#6217) This will enable TFRelu and TFRelu6 canonicalization Signed-off-by: SaeHie Park --- 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 80c5fca..ba1c4e7 100644 --- a/compiler/moco-tf/src/Canonicalizer.cpp +++ b/compiler/moco-tf/src/Canonicalizer.cpp @@ -29,6 +29,8 @@ #include "Canonicalization/IdentityCanonicalizer.h" #include "Canonicalization/MaxPoolCanonicalizer.h" #include "Canonicalization/MulCanonicalizer.h" +#include "Canonicalization/ReluCanonicalizer.h" +#include "Canonicalization/Relu6Canonicalizer.h" #include "Canonicalization/ReshapeCanonicalizer.h" #include "Dialect/TFDialect.h" @@ -78,6 +80,8 @@ 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()); phase.emplace_back(stdex::make_unique()); /* TRANSFORM DECLARATION END */ -- 2.7.4