From a27d574f827dcb3122bca89165eaeee91f189212 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: Wed, 17 Jul 2019 10:45:32 +0900 Subject: [PATCH] [moco/tf] ResolveFBN for only active nodes (#4311) This will change ResolveFusedBatchNorm transformation to iterate only active nodes Signed-off-by: SaeHie Park --- contrib/moco-tf/src/Transforms/ResolveFusedBatchNorm.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/contrib/moco-tf/src/Transforms/ResolveFusedBatchNorm.cpp b/contrib/moco-tf/src/Transforms/ResolveFusedBatchNorm.cpp index 37074af..a2194c1 100644 --- 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(node)) { -- 2.7.4