Fix Importer bug (#1541)

Found when testing 1 op models: `Im2Col` Layer uses the convolution parameter, as does
the `Deconvolution` (or `TransposedConvolution` layer). This resolves the issue.

Signed-off-by: Andrei Shedko <a.shedko@partner.samsung.com>

diff --git a/contrib/nnc/passes/caffe_frontend/caffe_model_visitor.cpp b/contrib/nnc/passes/caffe_frontend/caffe_model_visitor.cpp
index 5083140..bc93ce0 100644 (file)
--- a/contrib/nnc/passes/caffe_frontend/caffe_model_visitor.cpp
+++ b/contrib/nnc/passes/caffe_frontend/caffe_model_visitor.cpp
@@ -43,7 +43,7 @@ void ModelVisitor::visit(const LayerParameter& lp)
   {
     processInputLayer(lp);
   }
-  else if (lp.has_convolution_param())
+  else if (lp.type() == "Convolution")
   {
     outputs = opCreator.createConv2D(inputs, params, lp.convolution_param());
   }