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

Add getOutputDims to 2d layers

Browse Source
This commit is contained in:
Matúš Námešný
2024-05-26 14:28:43 +02:00
parent 78a0fd0baf
commit cba177e417
9 changed files with 37 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
include/layers/avg_pooling.cuh
View File

@@ -6,7 +6,7 @@
namespace CUDANet::Layers {
class AvgPooling2d : public SequentialLayer {
class AvgPooling2d : public SequentialLayer, public TwoDLayer {
public:
AvgPooling2d(
dim2d inputSize,
@@ -33,6 +33,8 @@ class AvgPooling2d : public SequentialLayer {
*/
int getInputSize();
dim2d getOutputDims();
private:
dim2d inputSize;
int nChannels;

4
include/layers/batch_norm.cuh
View File

@@ -8,7 +8,7 @@
namespace CUDANet::Layers {
class BatchNorm2d : public WeightedLayer {
class BatchNorm2d : public WeightedLayer, public TwoDLayer {
public:
BatchNorm2d(dim2d inputSize, int inputChannels, float epsilon, ActivationType activationType);
@@ -64,6 +64,8 @@ class BatchNorm2d : public WeightedLayer {
*/
int getInputSize();
dim2d getOutputDims();
private:
dim2d inputSize;

4
include/layers/conv2d.cuh
View File

@@ -13,7 +13,7 @@ namespace CUDANet::Layers {
* @brief 2D convolutional layer
*
*/
class Conv2d : public WeightedLayer {
class Conv2d : public WeightedLayer, public TwoDLayer {
public:
/**
* @brief Construct a new Conv 2d layer
@@ -102,6 +102,8 @@ class Conv2d : public WeightedLayer {
return paddingSize;
}
dim2d getOutputDims();
private:
// Inputs
dim2d inputSize;

9
include/layers/layer.cuh
View File

@@ -11,6 +11,15 @@ typedef std::pair<int, int> dim2d;
namespace CUDANet::Layers {
class TwoDLayer {
public:
virtual dim2d getOutputDims() = 0;
};
/**
* @brief Basic Sequential Layer
*

4
include/layers/max_pooling.cuh
View File

@@ -6,7 +6,7 @@
namespace CUDANet::Layers {
class MaxPooling2d : public SequentialLayer {
class MaxPooling2d : public SequentialLayer, public TwoDLayer {
public:
MaxPooling2d(
dim2d inputSize,
@@ -33,6 +33,8 @@ class MaxPooling2d : public SequentialLayer {
*/
int getInputSize();
dim2d getOutputDims();
private:
dim2d inputSize;
int nChannels;