MCPcopy Create free account
hub / github.com/apache/singa / Setup

Method Setup

src/model/layer/dropout.cc:27–31  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

25RegisterLayerClass(singacuda_dropout, Dropout);
26RegisterLayerClass(singacl_dropout, Dropout);
27void Dropout::Setup(const Shape& in_sample, const LayerConf& conf) {
28 Layer::Setup(in_sample, conf);
29 dropout_ratio_ = conf.dropout_conf().dropout_ratio();
30 out_sample_shape_= in_sample;
31}
32
33const Tensor Dropout::Forward(int flag, const Tensor& input) {
34 Tensor out;

Callers

nothing calls this directly

Calls 2

SetupFunction · 0.85
dropout_ratioMethod · 0.80

Tested by

no test coverage detected