From f71972df0c777797241dfa08aac3176bd356e30f 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, 2 Aug 2019 17:10:03 +0900 Subject: [PATCH] [moco-tf] Enable ResolveReshapeWildcardDim transform (#6154) This commit registers and enables ResolveReshapeWildcardDim 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 9001948..baf3deb 100644 --- a/compiler/moco-tf/src/TFOptimizer.cpp +++ b/compiler/moco-tf/src/TFOptimizer.cpp @@ -44,6 +44,7 @@ void TFOptimizer::optimize(loco::Graph *g) const // 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 */ moco::tf::PhaseRunner phase_runner{g}; diff --git a/compiler/moco-tf/src/Transforms.h b/compiler/moco-tf/src/Transforms.h index 3956245..accde2b 100644 --- a/compiler/moco-tf/src/Transforms.h +++ b/compiler/moco-tf/src/Transforms.h @@ -26,6 +26,7 @@ #include "Transforms/RemoveForwardNodeTransform.h" #include "Transforms/ReorderDecodeTransform.h" #include "Transforms/ResolveFusedBatchNorm.h" +#include "Transforms/ResolveReshapeWildcardDim.h" #include "Transforms/SimplifyDomainConversionTransform.h" #endif // __MOCO_TF_TRANSFORMS_H__ -- 2.7.4