3dconv to be tested

2017-06-29 08:47:29 +00:00
parent cfa43a8580
commit ffdb70abee
3 changed files with 189 additions and 2 deletions
@@ -9,7 +9,7 @@ cuda_add_library(kernels SHARED src/kernels/activation_elu.cu)

 include_directories(${CMAKE_CURRENT_SOURCE_DIR}/include ${CUDA_INCLUDE_DIRS})
 add_library(tkDNN SHARED src/Layer.cpp src/LayerWgs.cpp 
-    src/Dense.cpp src/Activation.cpp src/Conv2d.cpp
+    src/Dense.cpp src/Activation.cpp src/Conv2d.cpp src/Conv3d
    src/Network.cpp src/utils.cpp)
 target_link_libraries(tkDNN kernels)