From a720264cb67260e6a50fb0f293d819ba4c2a9688 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=84=B8=ED=9D=AC/On-Device=20Lab=28SR=29/Princip?= =?utf8?q?al=20Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Thu, 25 Jul 2019 13:26:56 +0900 Subject: [PATCH] [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 --- compiler/moco-tf/src/Transforms/FixPaddingTransform.cpp | 2 +- compiler/moco-tf/src/Transforms/FixShapeTransform.cpp | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/compiler/moco-tf/src/Transforms/FixPaddingTransform.cpp b/compiler/moco-tf/src/Transforms/FixPaddingTransform.cpp index e7d520b..6af16e9 100644 --- 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 --- 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) \ -- 2.7.4