Migrate Dense layer

2025-11-06 01:34:22 +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/cudanet.cuh
+++ b/include/cudanet.cuh
@@ -8,15 +8,15 @@
 #include "pooling.cuh"

 // Layers
-#include "activation.cuh"
+#include "activation.hpp"
 #include "add.cuh"
 #include "avg_pooling.cuh"
 #include "batch_norm.cuh"
 #include "concat.cuh"
 #include "conv2d.cuh"
-#include "dense.cuh"
+#include "dense.hpp"
 #include "input.cuh"
-#include "layer.cuh"
+#include "layer.hpp"
 #include "max_pooling.cuh"
 #include "output.cuh"