conv_edge.h

#ifndef CONV_EDGE_H_
#define CONV_EDGE_H_
#include "edge_with_weight.h"

class ConvEdge : public EdgeWithWeight {
 public:
  ConvEdge(const config::Edge& edge_config);
  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 void ComputeOuter(Matrix& input, Matrix& deriv_output);

  virtual void SetTiedTo(Edge* e);
  virtual void DisplayWeights();
  virtual int GetNumModules() const { return num_modules_; }
  virtual void SetImageSize(int image_size);
 
  int GetKernelSize() const { return kernel_size_; }
  int GetStride() const { return stride_; }
  int GetPadding() const { return padding_; }
  int GetPartialSum() const { return partial_sum_; }
  bool GetSharedBias() const { return shared_bias_; }

 private:
  void AllocateMemoryBprop();
  void AllocateMemoryFprop();

  Matrix grad_weights_partial_sum_;
  int kernel_size_, stride_, padding_, partial_sum_;
  bool shared_bias_;
};
#endif