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

Unify 2d layer naming

Browse Source
This commit is contained in:
Matúš Námešný
2024-05-20 16:23:58 +02:00
parent 74098b24e3
commit 6dca8ccd3c
12 changed files with 41 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
include/layers/avg_pooling.cuh
View File

@@ -6,16 +6,16 @@
namespace CUDANet::Layers {
class AvgPooling2D : public SequentialLayer {
class AvgPooling2d : public SequentialLayer {
public:
AvgPooling2D(
AvgPooling2d(
dim2d inputSize,
int nChannels,
dim2d poolingSize,
dim2d stride,
ActivationType activationType
);
~AvgPooling2D();
~AvgPooling2d();
float* forward(const float* d_input);

6
include/layers/batch_norm.cuh
View File

@@ -8,11 +8,11 @@
namespace CUDANet::Layers {
class BatchNorm2D : public WeightedLayer {
class BatchNorm2d : public WeightedLayer {
public:
BatchNorm2D(dim2d inputSize, int inputChannels, float epsilon, ActivationType activationType);
BatchNorm2d(dim2d inputSize, int inputChannels, float epsilon, ActivationType activationType);
~BatchNorm2D();
~BatchNorm2d();
/**
* @brief Compute the forward pass of the batchnorm layer

6
include/layers/max_pooling.cuh
View File

@@ -6,16 +6,16 @@
namespace CUDANet::Layers {
class MaxPooling2D : public SequentialLayer {
class MaxPooling2d : public SequentialLayer {
public:
MaxPooling2D(
MaxPooling2d(
dim2d inputSize,
int nChannels,
dim2d poolingSize,
dim2d stride,
ActivationType activationType
);
~MaxPooling2D();
~MaxPooling2d();
float* forward(const float* d_input);