[moco/tf] Add fix for shape and padding (#4188)

This will add fix for shape and padding after TF transformations that can add new nodes with empty shape and padding

Signed-off-by: SaeHie Park <saehie.park@samsung.com>

diff --git a/contrib/moco-tf/src/TFOptimizer.cpp b/contrib/moco-tf/src/TFOptimizer.cpp
index 5071e8d..9001948 100644 (file)
--- a/contrib/moco-tf/src/TFOptimizer.cpp
+++ b/contrib/moco-tf/src/TFOptimizer.cpp
@@ -40,6 +40,10 @@ void TFOptimizer::optimize(loco::Graph *g) const
   {
     phase.emplace_back(stdex::make_unique<moco::tf::FuseBinaryIntoPreceding>());
   }
+  // 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<moco::tf::FixShapeTransform>());
+  phase.emplace_back(stdex::make_unique<moco::tf::FixPaddingTransform>());
   /* TRANSFORM DECLARATION END */
 
   moco::tf::PhaseRunner<moco::tf::PhaseStrategy::Saturate> phase_runner{g};