Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-05 17:34:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Migrate Dense layer

Browse Source
This commit is contained in:
Matúš Námešný
2024-09-08 13:36:53 +02:00
parent 0dca8348bd
commit 75475790ac
24 changed files with 147 additions and 106 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/cuda/layers/test_activation.cu
View File

@@ -1,4 +1,4 @@
#include "activation.cuh"
#include "activation.hpp"
#include <gtest/gtest.h>
#include <cuda_runtime.h>
#include <vector>

2
test/cuda/layers/test_batch_norm.cu
View File

@@ -3,7 +3,7 @@
#include <vector>
#include "activation.cuh"
#include "activation.hpp"
#include "batch_norm.cuh"
class BatchNormLayerTest : public ::testing::Test {

4
test/cuda/layers/test_dense.cu
View File

@@ -3,8 +3,8 @@
#include <iostream>
#include "activation.cuh"
#include "dense.cuh"
#include "activation.hpp"
#include "dense.hpp"
class DenseLayerTest : public ::testing::Test {
protected:

2
test/model/test_model.cpp
View File

@@ -1,7 +1,7 @@
#include <gtest/gtest.h>
#include "conv2d.cuh"
#include "dense.cuh"
#include "dense.hpp"
#include "max_pooling.cuh"
#include "model.hpp"