[moco/tf] ResolveFBN for only active nodes (#4311)

This will change ResolveFusedBatchNorm transformation to iterate only active nodes

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

diff --git a/contrib/moco-tf/src/Transforms/ResolveFusedBatchNorm.cpp b/contrib/moco-tf/src/Transforms/ResolveFusedBatchNorm.cpp
index 37074af..a2194c1 100644 (file)
--- a/contrib/moco-tf/src/Transforms/ResolveFusedBatchNorm.cpp
+++ b/contrib/moco-tf/src/Transforms/ResolveFusedBatchNorm.cpp
@@ -241,7 +241,7 @@ namespace tf
 
 bool ResolveFusedBatchNorm::run(loco::Graph *graph)
 {
-  for (auto node : loco::all_nodes(graph))
+  for (auto node : loco::active_nodes(loco::output_nodes(graph)))
   {
     if (as<moco::tf::TFFusedBatchNorm>(node))
     {