add split layer insertion tests; move split insertion code to util file

1 files changed, 21 insertions, 0 deletions

diff --git a/include/caffe/util/insert_splits.hpp b/include/caffe/util/insert_splits.hpp
new file mode 100644
index 00000000..f8aae288
--- /dev/null
+++ b/include/caffe/util/insert_splits.hpp
@@ -0,0 +1,21 @@
+// Copyright 2014 Jeff Donahue
+
+#ifndef _CAFFE_UTIL_INSERT_SPLITS_HPP_
+#define _CAFFE_UTIL_INSERT_SPLITS_HPP_
+
+#include "caffe/proto/caffe.pb.h"
+
+using std::string;
+
+namespace caffe {
+
+// Copy NetParameters with SplitLayers added to replace any shared bottom
+// blobs with unique bottom blobs provided by the SplitLayer.
+void insert_splits(const NetParameter& param, NetParameter* param_split);
+
+void configure_split_layer(const string& blob_name,
+    const int split_count, LayerConnection* split_layer_connection);
+
+}  // namespace caffe
+
+#endif  // CAFFE_UTIL_INSERT_SPLITS_HPP_