Migrate Dense layer

2025-11-05 17:34:21 +00:00 · 2024-09-08 13:36:53 +02:00
parent 0dca8348bd
commit 75475790ac
24 changed files with 147 additions and 106 deletions
--- a/include/layers/activation.hpp
+++ b/include/layers/activation.hpp
@@ -48,6 +48,8 @@ class Activation {
    ActivationType activationType;
    int length;

+    void activateCPU(float* input);
+
 #ifdef USE_CUDA
    int gridSize;

@@ -58,10 +60,7 @@ class Activation {

    void initCUDA();
    void delCUDA();
-#else
-    void activateCPU(float* input);
 #endif
-
 };