Migrate output layer

include/cudanet.cuh

@@ -19,7 +19,7 @@
 #include "input.hpp"
 #include "layer.hpp"
 #include "max_pooling.hpp"
-#include "output.cuh"
+#include "output.hpp"
 
 // Models
 #include "model.hpp"

include/layers/output.hpp

@@ -46,6 +46,12 @@ class Output : public SequentialLayer {
   private:
     int    inputSize;
     float* h_output;
+
+    float* forwardCPU(const float* input);
+
+#ifdef USE_CUDA
+    float* forwardCUDA(const float* input);
+#endif
 };
 
 }  // namespace CUDANet::Layers

include/model/model.hpp

@@ -8,7 +8,7 @@
 #include "input.hpp"
 #include "layer.hpp"
 #include "module.hpp"
-#include "output.cuh"
+#include "output.hpp"
 
 namespace CUDANet {