MCPcopy Create free account
hub / github.com/BVLC/caffe / RNNLayerTest

Method RNNLayerTest

src/caffe/test/test_rnn_layer.cpp:21–39  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

19
20 protected:
21 RNNLayerTest() : num_output_(7) {
22 blob_bottom_vec_.push_back(&blob_bottom_);
23 blob_bottom_vec_.push_back(&blob_bottom_cont_);
24 blob_top_vec_.push_back(&blob_top_);
25
26 ReshapeBlobs(1, 3);
27
28 layer_param_.mutable_recurrent_param()->set_num_output(num_output_);
29 FillerParameter* weight_filler =
30 layer_param_.mutable_recurrent_param()->mutable_weight_filler();
31 weight_filler->set_type("gaussian");
32 weight_filler->set_std(0.2);
33 FillerParameter* bias_filler =
34 layer_param_.mutable_recurrent_param()->mutable_bias_filler();
35 bias_filler->set_type("gaussian");
36 bias_filler->set_std(0.1);
37
38 layer_param_.set_phase(TEST);
39 }
40
41 void ReshapeBlobs(int num_timesteps, int num_instances) {
42 blob_bottom_.Reshape(num_timesteps, num_instances, 3, 2);

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected