Migrate max pooling

2025-11-06 01:34:22 +00:00 · 2024-09-09 21:48:36 +02:00
parent 8a5d556b7e
commit a0665fb05c
6 changed files with 71 additions and 33 deletions
--- a/test/model/test_model.cpp
+++ b/test/model/test_model.cpp
@@ -2,7 +2,7 @@

 #include "conv2d.cuh"
 #include "dense.hpp"
-#include "max_pooling.cuh"
+#include "max_pooling.hpp"
 #include "model.hpp"

 class ModelTest : public ::testing::Test {