[nnc] Fix broken importing of Scale Caffe layer (#2704)
authorСергей Баранников/AI Tools Lab /SRR/Engineer/삼성전자 <s.barannikov@samsung.com>
Mon, 17 Dec 2018 18:51:40 +0000 (21:51 +0300)
committerEfimov Alexander/AI Tools Lab/./Samsung Electronics <a.efimov@samsung.com>
Mon, 17 Dec 2018 18:51:40 +0000 (21:51 +0300)
Fix broken importing of Scale Caffe layer when --debug_transpose is passed to the compiler.

Signed-off-by: Sergei Barannikov <s.barannikov@samsung.com>
contrib/nnc/passes/caffe_frontend/caffe_op_creator.cpp

index b9731a7..58c7c27 100644 (file)
@@ -481,7 +481,7 @@ CaffeOpCreator::convertScale(const caffe::LayerParameter& layer,
     result = createOp<ops::BiasAddOp>(layer.name() + ".bias", result->getOutput(0), bias_weights);
   }
 
-  return {result->getOutput(0)};
+  return {convertMIRToCaffe(result->getOutput(0))};
 }
 
 void CaffeOpCreator::checkBatchNorm(const caffe::LayerParameter& layer,