Migrate concat layer

2025-11-06 01:34:22 +00:00 · 2024-09-09 22:16:22 +02:00
parent a0665fb05c
commit fe7c16ac36
6 changed files with 80 additions and 42 deletions
--- a/include/cudanet.cuh
+++ b/include/cudanet.cuh
@@ -13,7 +13,7 @@
 #include "add.hpp"
 #include "avg_pooling.hpp"
 #include "batch_norm.cuh"
-#include "concat.cuh"
+#include "concat.hpp"
 #include "conv2d.cuh"
 #include "dense.hpp"
 #include "input.cuh"