@@ -35,7 +35,8 @@ class AgentLayer : public Layer {
35
35
36
36
~AgentLayer () {}
37
37
38
- bool init (const LayerMap& layerMap, const ParameterMap& parameterMap);
38
+ bool init (const LayerMap& layerMap,
39
+ const ParameterMap& parameterMap) override ;
39
40
40
41
// if *numSamples* set,
41
42
// real layer output will only use first *numSamples* rows
@@ -44,8 +45,8 @@ class AgentLayer : public Layer {
44
45
numSamples_ = numSamples;
45
46
}
46
47
47
- void forward (PassType passType);
48
- void backward (const UpdateCallback& callback = nullptr ) {}
48
+ void forward (PassType passType) override ;
49
+ void backward (const UpdateCallback& callback = nullptr ) override {}
49
50
};
50
51
51
52
/* *
@@ -56,8 +57,8 @@ class SequenceAgentLayer : public AgentLayer {
56
57
explicit SequenceAgentLayer (const LayerConfig& config) : AgentLayer(config) {}
57
58
~SequenceAgentLayer () {}
58
59
59
- void forward (PassType passType);
60
- void backward (const UpdateCallback& callback = nullptr ) {}
60
+ void forward (PassType passType) override ;
61
+ void backward (const UpdateCallback& callback = nullptr ) override {}
61
62
};
62
63
63
64
/* *
@@ -78,7 +79,8 @@ class GatherAgentLayer : public Layer {
78
79
79
80
virtual ~GatherAgentLayer () {}
80
81
81
- bool init (const LayerMap& layerMap, const ParameterMap& parameterMap);
82
+ bool init (const LayerMap& layerMap,
83
+ const ParameterMap& parameterMap) override ;
82
84
83
85
// call before addRealLayer
84
86
void copyIdAndSequenceInfo (const Argument& input,
@@ -88,8 +90,8 @@ class GatherAgentLayer : public Layer {
88
90
// add one real layer, can call many times
89
91
void addRealLayer (LayerPtr layer) { realLayers_.push_back (layer); }
90
92
91
- void forward (PassType passType);
92
- void backward (const UpdateCallback& callback);
93
+ void forward (PassType passType) override ;
94
+ void backward (const UpdateCallback& callback) override ;
93
95
};
94
96
95
97
/* *
@@ -133,7 +135,8 @@ class ScatterAgentLayer : public Layer {
133
135
134
136
virtual ~ScatterAgentLayer () {}
135
137
136
- bool init (const LayerMap& layerMap, const ParameterMap& parameterMap);
138
+ bool init (const LayerMap& layerMap,
139
+ const ParameterMap& parameterMap) override ;
137
140
138
141
/* *
139
142
* @brief set real layer in generation
@@ -182,8 +185,8 @@ class ScatterAgentLayer : public Layer {
182
185
numSequences_ = numSequences;
183
186
}
184
187
185
- void forward (PassType passType);
186
- void backward (const UpdateCallback& callback);
188
+ void forward (PassType passType) override ;
189
+ void backward (const UpdateCallback& callback) override ;
187
190
};
188
191
189
192
/* *
0 commit comments