[neurun] Mark FromCommon for all of ifm indexes in concat (#2185)
author김수진/동작제어Lab(SR)/Engineer/삼성전자 <sjsujin.kim@samsung.com>
Tue, 7 Aug 2018 10:43:35 +0000 (19:43 +0900)
committer오형석/동작제어Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com>
Tue, 7 Aug 2018 10:43:35 +0000 (19:43 +0900)
This commit marks FromCommon for all of ifm indexes in concat.

It'was a bug.

Signed-off-by: sjsujinkim <sjsujin.kim@samsung.com>
runtimes/neurun/src/compilation.cc

index 5c52c86..eff8b32 100644 (file)
@@ -487,9 +487,8 @@ void TensorMarker::visit(const ::internal::tflite::op::Concat::Node &node)
   for (auto ind : param.ifm_indexes)
   {
     mark(ind);
+    markFromCommon(node, ind);
   }
-
-  markFromCommon(node, param.ifm_indexes[0]);
 }
 
 void TensorMarker::visit(const ::internal::tflite::op::FullyConnected::Node &node)