Refactor layers

include/layers/layer.cuh

@@ -0,0 +1,91 @@
+
+#ifndef CUDANET_I_LAYER_H
+#define CUDANET_I_LAYER_H
+
+#include <vector>
+
+namespace CUDANet::Layers {
+
+/**
+ * @brief Padding types
+ *
+ * SAME: Zero padding such that the output size is the same as the input
+ * VALID: No padding
+ *
+ */
+enum Padding { SAME, VALID };
+
+/**
+ * @brief Basic Sequential Layer
+ * 
+ */
+class SequentialLayer {
+  public:
+    /**
+     * @brief Destroy the Sequential Layer
+     * 
+     */
+    virtual ~SequentialLayer() {};
+
+    /**
+     * @brief Forward propagation virtual function
+     * 
+     * @param input Device pointer to the input
+     * @return float* Device pointer to the output
+     */
+    virtual float* forward(const float* input) = 0;
+};
+
+/**
+ * @brief Base class for layers with weights and biases
+ */
+class WeightedLayer : public SequentialLayer {
+  public:
+    /**
+     * @brief Destroy the ILayer object
+     *
+     */
+    virtual ~WeightedLayer() {};
+
+    /**
+     * @brief Virtual function for forward pass
+     *
+     * @param input (Device) Pointer to the input
+     * @return float* Device pointer to the output
+     */
+    virtual float* forward(const float* input) = 0;
+
+    /**
+     * @brief Virtual function for setting weights
+     *
+     * @param weights Pointer to the weights
+     */
+    virtual void setWeights(const float* weights) = 0;
+
+    /**
+     * @brief Virtual function for setting biases
+     *
+     * @param biases Pointer to the biases
+     */
+    virtual void setBiases(const float* biases) = 0;
+
+  private:
+    /**
+     * @brief Initialize the weights
+     */
+    virtual void initializeWeights() = 0;
+
+    /**
+     * @brief Initialize the biases
+     */
+    virtual void initializeBiases() = 0;
+
+    /**
+     * @brief Copy the weights and biases to the device
+     */
+    virtual void toCuda() = 0;
+};
+
+}  // namespace CUDANet::Layers
+
+#endif  // CUDANET_I_LAYERH