| 33 | /// \copydoc Layer::Setup(const LayerConf&); |
| 34 | void Setup(const Shape& in_sample, const LayerConf& conf) override; |
| 35 | const Shape GetOutputSampleShape() const override { |
| 36 | CHECK(out_sample_shape_.size()) << "You may haven't call Setup()"; |
| 37 | return out_sample_shape_; |
| 38 | } |
| 39 | /// \copydoc Layer::Forward(int flag, const Tensor&) |
| 40 | const Tensor Forward(int flag, const Tensor& input) override; |
| 41 |