Support Split operation in caffe frontend (#1475)

Support Split operation in caffe importer

Signed-off-by: Efimov Alexander <a.efimov@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 dc3d773..5083140 100644 (file)
--- a/contrib/nnc/passes/caffe_frontend/caffe_model_visitor.cpp
+++ b/contrib/nnc/passes/caffe_frontend/caffe_model_visitor.cpp
@@ -81,7 +81,15 @@ void ModelVisitor::visit(const LayerParameter& lp)
   }
   else if (lp.has_dropout_param())
   {
-    outputs = opCreator.createDropout(inputs, params,lp.dropout_param());
+    outputs = opCreator.createDropout(inputs, params, lp.dropout_param());
+  }
+  else if (lp.type() == "Split")
+  {
+    INode *prev = opsForBlobsTheyOutput[lp.bottom(0)];
+    for (int i = 0; i < lp.top_size(); ++i)
+    {
+      opsForBlobsTheyOutput[lp.top(i)] = prev;
+    }
   }
   else
   {