Skip to content

Commit dfedd21

Browse files
committed
Add layer names in subgraph of RNNv2
1 parent 3d3f6b1 commit dfedd21

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/caffe/layers/rnn_v2_layer.cpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -282,12 +282,14 @@ void RNNv2Layer<Dtype>::FillUnrolledNet(
282282
h_T_copy_param->CopyFrom(split_param);
283283
h_T_copy_param->add_bottom(layer_name_prefix + recur_name_prefix[0] + format_int(this->T_));
284284
h_T_copy_param->add_top(layer_name_prefix + recur_name_prefix[0] + "T");
285+
h_T_copy_param->set_name(layer_name_prefix + recur_name_prefix[0] + "T");
285286
}
286287
{
287288
LayerParameter *c_T_copy_param = net_param->add_layer();
288289
c_T_copy_param->CopyFrom(split_param);
289290
c_T_copy_param->add_bottom(layer_name_prefix + recur_name_prefix[1] + format_int(this->T_));
290291
c_T_copy_param->add_top(layer_name_prefix + recur_name_prefix[1] + "T");
292+
c_T_copy_param->set_name(layer_name_prefix + recur_name_prefix[1] + "T");
291293
}
292294
net_param->add_layer()->CopyFrom(output_concat_layer);
293295
}
@@ -345,6 +347,7 @@ void RNNv2Layer<Dtype>::LayerSetUp(const vector<Blob<Dtype> *> &bottom,
345347

346348
LayerParameter *input_layer_param = net_param.add_layer();
347349
input_layer_param->set_type("Input");
350+
input_layer_param->set_name("Input");
348351
InputParameter *input_param = input_layer_param->mutable_input_param();
349352
input_layer_param->add_top("x");
350353
BlobShape input_shape;

0 commit comments

Comments
 (0)