| 260 | class ConvModule : public Module { |
| 261 | public: |
| 262 | ConvModule(const NN::ConvParameters& parameters) { |
| 263 | mParameter = parameters; |
| 264 | if (nullptr != mParameter.bias) { |
| 265 | addParameter(mParameter.bias); |
| 266 | } |
| 267 | if (nullptr != mParameter.weight) { |
| 268 | addParameter(mParameter.weight); |
| 269 | } |
| 270 | setName(parameters.name); |
| 271 | setType("Conv"); |
| 272 | } |
| 273 | NN::ConvParameters& convParameters() { |
| 274 | return mParameter; |
| 275 | } |