Migrate avg pooling

2025-11-06 01:34:22 +00:00 · 2024-09-09 21:36:13 +02:00
parent 75475790ac
commit 76e5225001
5 changed files with 129 additions and 53 deletions
--- a/test/cuda/layers/test_avg_pooling.cu
+++ b/test/cuda/layers/test_avg_pooling.cu
@@ -3,7 +3,7 @@

 #include <vector>

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

 class AvgPoolingLayerTest : public ::testing::Test {
  protected: