maxpool_edge.h

#ifndef MAXPOOL_EDGE_H_
#define MAXPOOL_EDGE_H_
#include "edge.h"

class MaxPoolEdge : public Edge {
 public:
  MaxPoolEdge(const config::Edge& edge_config);
  virtual void SetTiedTo(Edge* e);
  virtual void AllocateMemory(bool fprop_only);
  virtual void ComputeUp(Matrix& input, Matrix& output, bool overwrite);
  virtual void ComputeDown(Matrix& deriv_output, Matrix& input,
                           Matrix& output, Matrix& deriv_input, bool overwrite);

  virtual int GetNumModules() const { return num_modules_; }
  virtual bool RequiresMemoryForDeriv() const { return true; }
  virtual void SetImageSize(int image_size);

  int GetKernelSize() const { return kernel_size_; }
  int GetStride() const { return stride_; }
  int GetPadding() const { return padding_; }

 private:
  int kernel_size_, stride_, padding_;
};
#endif