[moco-tf] Run for active nodes for FixPad and FixShape (#5862)

This will fix to run for active nodes from outputs for FixPadding and FixShape transformations

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

diff --git a/compiler/moco-tf/src/Transforms/FixPaddingTransform.cpp b/compiler/moco-tf/src/Transforms/FixPaddingTransform.cpp
index e7d520b..6af16e9 100644 (file)
--- a/compiler/moco-tf/src/Transforms/FixPaddingTransform.cpp
+++ b/compiler/moco-tf/src/Transforms/FixPaddingTransform.cpp
@@ -485,7 +485,7 @@ namespace tf
 bool FixPaddingTransform::run(loco::Graph *graph)
 {
   bool changed = false;
-  for (auto node : loco::all_nodes(graph))
+  for (auto node : loco::active_nodes(loco::output_nodes(graph)))
   {
 // clang-format off
 #define CANONICAL_NODE(TYPE_NAME)                       \

diff --git a/compiler/moco-tf/src/Transforms/FixShapeTransform.cpp b/compiler/moco-tf/src/Transforms/FixShapeTransform.cpp
index 415ef62..69003af 100644 (file)
--- a/compiler/moco-tf/src/Transforms/FixShapeTransform.cpp
+++ b/compiler/moco-tf/src/Transforms/FixShapeTransform.cpp
@@ -765,7 +765,7 @@ namespace tf
 bool FixShapeTransform::run(loco::Graph *graph)
 {
   bool changed = false;
-  for (auto node : loco::all_nodes(graph))
+  for (auto node : loco::active_nodes(loco::output_nodes(graph)))
   {
 // clang-format off
 #define CANONICAL_NODE(TYPE_NAME)                     \