From 7e5d4a9753a67cd5d78809f2480a36e128a1bfa1 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=B2=9C=EA=B5=90/On-Device=20Lab=28SR=29/Enginee?= =?utf8?q?r/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Fri, 9 Aug 2019 10:59:11 +0900 Subject: [PATCH] [moco-tf] Enable ResolveConstantShape (#6423) This commit enables ResolveConstantShape transform Signed-off-by: Cheongyo Bahk --- compiler/moco-tf/src/TFOptimizer.cpp | 1 + compiler/moco-tf/src/Transforms.h | 1 + 2 files changed, 2 insertions(+) diff --git a/compiler/moco-tf/src/TFOptimizer.cpp b/compiler/moco-tf/src/TFOptimizer.cpp index 4789df2..d295756 100644 --- a/compiler/moco-tf/src/TFOptimizer.cpp +++ b/compiler/moco-tf/src/TFOptimizer.cpp @@ -43,6 +43,7 @@ void TFOptimizer::optimize(loco::Graph *g) const // Fix shape and pad for added nodes doing above transformations // TODO need to merge or remove the ones in importer phase.emplace_back(stdex::make_unique()); + phase.emplace_back(stdex::make_unique()); phase.emplace_back(stdex::make_unique()); /* TRANSFORM DECLARATION END */ diff --git a/compiler/moco-tf/src/Transforms.h b/compiler/moco-tf/src/Transforms.h index cfc607d..b108688 100644 --- a/compiler/moco-tf/src/Transforms.h +++ b/compiler/moco-tf/src/Transforms.h @@ -24,6 +24,7 @@ #include "Transforms/RemoveDeadNodeTransform.h" #include "Transforms/RemoveForwardNodeTransform.h" #include "Transforms/ReorderDecodeTransform.h" +#include "Transforms/ResolveConstantShape.h" #include "Transforms/ResolveFusedBatchNorm.h" #include "Transforms/ResolveReshapeWildcardDim.h" #include "Transforms/SimplifyDomainConversionTransform.h" -- 2.7.4