Add getOutputDims to 2d layers

include/layers/avg_pooling.cuh

@@ -6,7 +6,7 @@
 
 namespace CUDANet::Layers {
 
-class AvgPooling2d : public SequentialLayer {
+class AvgPooling2d : public SequentialLayer, public TwoDLayer {
   public:
     AvgPooling2d(
         dim2d           inputSize,
@@ -33,6 +33,8 @@ class AvgPooling2d : public SequentialLayer {
      */
     int getInputSize();
 
+    dim2d getOutputDims();
+
   private:
     dim2d inputSize;
     int  nChannels;

include/layers/batch_norm.cuh

@@ -8,7 +8,7 @@
 
 namespace CUDANet::Layers {
 
-class BatchNorm2d : public WeightedLayer {
+class BatchNorm2d : public WeightedLayer, public TwoDLayer {
   public:
     BatchNorm2d(dim2d inputSize, int inputChannels, float epsilon, ActivationType activationType);
 
@@ -64,6 +64,8 @@ class BatchNorm2d : public WeightedLayer {
      */
     int getInputSize();
 
+    dim2d getOutputDims();
+
   private:
 
     dim2d inputSize;

include/layers/conv2d.cuh

@@ -13,7 +13,7 @@ namespace CUDANet::Layers {
  * @brief 2D convolutional layer
  *
  */
-class Conv2d : public WeightedLayer {
+class Conv2d : public WeightedLayer, public TwoDLayer {
   public:
     /**
      * @brief Construct a new Conv 2d layer
@@ -102,6 +102,8 @@ class Conv2d : public WeightedLayer {
         return paddingSize;
     }
 
+    dim2d getOutputDims();
+
   private:
     // Inputs
     dim2d inputSize;

include/layers/layer.cuh

@@ -11,6 +11,15 @@ typedef std::pair<int, int> dim2d;
 
 namespace CUDANet::Layers {
 
+
+class TwoDLayer {
+
+  public:
+    virtual dim2d getOutputDims() = 0;
+
+};
+
+
 /**
  * @brief Basic Sequential Layer
  *

include/layers/max_pooling.cuh

@@ -6,7 +6,7 @@
 
 namespace CUDANet::Layers {
 
-class MaxPooling2d : public SequentialLayer {
+class MaxPooling2d : public SequentialLayer, public TwoDLayer {
   public:
     MaxPooling2d(
         dim2d          inputSize,
@@ -33,6 +33,8 @@ class MaxPooling2d : public SequentialLayer {
      */
     int getInputSize();
 
+    dim2d getOutputDims();
+
   private:
     dim2d inputSize;
     int   nChannels;