[moco/tf] revisit graph cleanup (#3940)
author박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Mon, 24 Jun 2019 07:16:17 +0000 (16:16 +0900)
committerGitHub Enterprise <noreply-CODE@samsung.com>
Mon, 24 Jun 2019 07:16:17 +0000 (16:16 +0900)
* [moco/tf] revisit graph cleanup

This will remove prepare and cleanup transformation in graph load method
- shape inference annotation can be reused after load with this

Signed-off-by: SaeHie Park <saehie.park@samsung.com>
* rollback mem leak comments

* undo remove prepare

contrib/moco/lib/frontend/tf/src/Frontend.cpp

index c2d5ff4..01e1bd4 100644 (file)
@@ -263,15 +263,10 @@ void transform_graph(loco::Graph *graph)
 {
   std::vector<std::unique_ptr<moco::tf::Transform>> prepare;
   std::vector<std::unique_ptr<moco::tf::Transform>> transforms;
-  std::vector<std::unique_ptr<moco::tf::Transform>> finalize;
 
   // Transforms that run only once for preparation and finalization
   {
-    // TODO add one time preparation when needed
-
-    // To remove frontend temporary annotations
-    finalize.emplace_back(stdex::make_unique<moco::tf::ClearAnnotTransform>());
-    // TODO add one time finalization when needed
+      // TODO add one time preparation when needed
   }
 
   // Transforms that run multiple times until there is no transform occured
@@ -310,12 +305,6 @@ void transform_graph(loco::Graph *graph)
     }
   }
 
-  // Run finalize to cleanup temporary annotations
-  for (auto &tr : finalize)
-  {
-    tr->run(graph);
-  }
-
   // validate graph
   assert(loco::valid(graph));
 }