get_split_blob_name returns a string to remove some verbosity

1 files changed, 10 insertions, 17 deletions

diff --git a/src/caffe/util/insert_splits.cpp b/src/caffe/util/insert_splits.cpp
index b8098844..eaf1f230 100644
--- a/src/caffe/util/insert_splits.cpp
+++ b/src/caffe/util/insert_splits.cpp
@@ -54,10 +54,8 @@ void insert_splits(const NetParameter& param, NetParameter* param_split) {
       const string& blob_name = layer_connection->bottom(j);
       const int split_count = blob_name_to_bottom_count[blob_name];
       if (split_count > 1) {
-        string split_blob_name;
-        get_split_blob_name(blob_name,
-            blob_name_to_bottom_split_idx[blob_name]++, &split_blob_name);
-        layer_connection->set_bottom(j, split_blob_name);
+        layer_connection->set_bottom(j, get_split_blob_name(blob_name,
+            blob_name_to_bottom_split_idx[blob_name]++));
       }
     }
     // Create split layer for any top blobs used by other layers as bottom
@@ -82,24 +80,19 @@ void configure_split_layer(const string& blob_name,
   split_layer_param->set_name(blob_name + "_split");
   split_layer_param->set_type("split");
   for (int k = 0; k < split_count; ++k) {
-    string split_blob_name;
-    get_split_blob_name(blob_name, k, &split_blob_name);
-    split_layer_connection->add_top(split_blob_name);
+    split_layer_connection->add_top(get_split_blob_name(blob_name, k));
   }
 }
 
-void get_split_blob_name(const string& blob_name, const int split_index,
-    string* split_blob_name) {
+string get_split_blob_name(const string& blob_name, const int split_index) {
   if (split_index == 0) {
-    *split_blob_name = blob_name;
-  } else {
-    const int suffix_max_length = 16;
-    char split_suffix[suffix_max_length];
-    const int suffix_length = snprintf(split_suffix, suffix_max_length,
-        "_split_%d", split_index);
-    CHECK_LT(suffix_length, suffix_max_length);
-    *split_blob_name = blob_name + split_suffix;
+    return blob_name;
   }
+  const int suffix_max_length = 16;
+  char split_suffix[suffix_max_length];
+  CHECK_LT(snprintf(split_suffix, suffix_max_length, "_split_%d", split_index),
+           suffix_max_length);
+  return blob_name + split_suffix;
 }
 
 }  // namespace caffe