Migrate input layer

include/cudanet.cuh

@@ -16,7 +16,7 @@
 #include "concat.hpp"
 #include "conv2d.cuh"
 #include "dense.hpp"
-#include "input.cuh"
+#include "input.hpp"
 #include "layer.hpp"
 #include "max_pooling.hpp"
 #include "output.cuh"

include/layers/input.hpp

@@ -40,16 +40,25 @@ class Input : public SequentialLayer {
      */
     int getOutputSize();
 
-        /**
+    /**
      * @brief Get input size
-     * 
+     *
      * @return int input size
      */
     int getInputSize();
 
   private:
-    int    inputSize;
+    int inputSize;
+
+    float* forwardCPU(const float* input);
+
+#ifdef USE_CUDA
     float* d_output;
+
+    float* forwardCUDA(const float* input);
+    void   initCUDA();
+    void   delCUDA();
+#endif
 };
 
 }  // namespace CUDANet::Layers

include/model/model.hpp

@@ -5,7 +5,7 @@
 #include <unordered_map>
 #include <vector>
 
-#include "input.cuh"
+#include "input.hpp"
 #include "layer.hpp"
 #include "module.hpp"
 #include "output.cuh"