Migrate max pooling

2025-11-05 17:34:21 +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/cuda/layers/test_max_pooling.cu
+++ b/test/cuda/layers/test_max_pooling.cu
@@ -3,7 +3,7 @@

 #include <vector>

-#include "max_pooling.cuh"
+#include "max_pooling.hpp"

 class MaxPoolingLayerTest : public ::testing::Test {
  protected:
--- 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 {