Migrate conv2d layer

2025-11-06 01:34:22 +00:00 · 2024-09-10 21:48:47 +02:00
parent 757584544c
commit 98ad4ac760
6 changed files with 180 additions and 123 deletions
--- a/test/cuda/layers/test_batch_norm.cu
+++ b/test/cuda/layers/test_batch_norm.cu
@@ -4,7 +4,7 @@
 #include <vector>

 #include "activation.hpp"
-#include "batch_norm.cuh"
+#include "batch_norm.hpp"

 class BatchNormLayerTest : public ::testing::Test {
  protected: