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

Implement BasicConv2d test

Browse Source
This commit is contained in:
Matúš Námešný
2024-06-02 14:40:46 +02:00
parent f5113cbac9
commit 7f745027cd
4 changed files with 296 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test/layers/test_conv2d.cu
View File

@@ -7,12 +7,12 @@
class Conv2dTest : public ::testing::Test {
protected:
shape2d inputSize;
shape2d inputSize;
int inputChannels;
shape2d kernelSize;
shape2d stride;
shape2d kernelSize;
shape2d stride;
int numFilters;
shape2d paddingSize;
shape2d paddingSize;
CUDANet::Layers::ActivationType activationType;
std::vector<float> input;
std::vector<float> kernels;