Abstract activation and implement softmax

This commit is contained in:
2024-03-17 18:37:15 +01:00
parent b1621819ca
commit 42d646750b
19 changed files with 370 additions and 205 deletions

View File

@@ -2,10 +2,11 @@ find_package(GTest REQUIRED)
include_directories(${GTEST_INCLUDE_DIRS})
add_executable(test_main
EXCLUDE_FROM_ALL
layers/test_dense.cu
layers/test_conv2d.cu
layers/test_input.cu
kernels/test_activations.cu
kernels/test_activation_functions.cu
kernels/test_padding.cu
kernels/test_matmul.cu
)