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

Rename dim2d to shape2d

Browse Source
This commit is contained in:
Matúš Námešný
2024-05-27 21:14:51 +02:00
parent 07d505a0e5
commit df47a31f36
22 changed files with 120 additions and 120 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test/layers/test_avg_pooling.cu
View File

@@ -7,11 +7,11 @@
class AvgPoolingLayerTest : public ::testing::Test {
protected:
dim2d inputSize;
shape2d inputSize;
int nChannels;
dim2d poolingSize;
dim2d stride;
dim2d padding;
shape2d poolingSize;
shape2d stride;
shape2d padding;
std::vector<float> input;
std::vector<float> expected;

2
test/layers/test_batch_norm.cu
View File

@@ -8,7 +8,7 @@
class BatchNormLayerTest : public ::testing::Test {
protected:
dim2d inputSize;
shape2d inputSize;
int nChannels;
std::vector<float> weights;
std::vector<float> biases;

8
test/layers/test_conv2d.cu
View File

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

8
test/layers/test_max_pooling.cu
View File

@@ -7,11 +7,11 @@
class MaxPoolingLayerTest : public ::testing::Test {
protected:
dim2d inputSize;
shape2d inputSize;
int nChannels;
dim2d poolingSize;
dim2d stride;
dim2d padding;
shape2d poolingSize;
shape2d stride;
shape2d padding;
std::vector<float> input;
std::vector<float> expected;

14
test/model/test_model.cu
View File

@@ -10,21 +10,21 @@ class ModelTest : public ::testing::Test {
CUDANet::Model *commonTestSetup(
bool setWeights = true,
dim2d inputSize = {6, 6},
shape2d inputSize = {6, 6},
int inputChannels = 2,
int outputSize = 3,
dim2d kernelSize = {3, 3},
dim2d stride = {1, 1},
shape2d kernelSize = {3, 3},
shape2d stride = {1, 1},
int numFilters = 2,
dim2d poolingSize = {2, 2},
dim2d poolingStride = {2, 2}
shape2d poolingSize = {2, 2},
shape2d poolingStride = {2, 2}
) {
CUDANet::Model *model =
new CUDANet::Model(inputSize, inputChannels, outputSize);
dim2d paddingSize = {0, 0};
shape2d paddingSize = {0, 0};
// Conv2d
CUDANet::Layers::Conv2d *conv2d = new CUDANet::Layers::Conv2d(
@@ -38,7 +38,7 @@ class ModelTest : public ::testing::Test {
model->addLayer("conv1", conv2d);
// maxpool2d
dim2d poolingInput = {
shape2d poolingInput = {
inputSize.first - kernelSize.first + 1,
inputSize.second - kernelSize.second + 1
};