Implement simple input layer

2025-11-06 01:34:22 +00:00 · 2024-03-12 21:16:46 +01:00
parent 9d91896f13
commit 708164e4d0
5 changed files with 82 additions and 1 deletions
--- a/test/CMakeLists.txt
+++ b/test/CMakeLists.txt
@@ -4,6 +4,7 @@ include_directories(${GTEST_INCLUDE_DIRS})
 add_executable(test_main 
    layers/test_dense.cu
    layers/test_conv2d.cu
+    layers/test_input.cu
    kernels/test_activations.cu
    kernels/test_padding.cu
 )