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

Add adaptive avg pooling

Browse Source
This commit is contained in:
Matúš Námešný
2024-05-30 17:17:31 +02:00
parent 9faf20876a
commit 8168f02f58
5 changed files with 143 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
include/layers/avg_pooling.cuh
View File

@@ -9,11 +9,11 @@ namespace CUDANet::Layers {
class AvgPooling2d : public SequentialLayer, public TwoDLayer {
public:
AvgPooling2d(
shape2d inputSize,
shape2d inputSize,
int nChannels,
shape2d poolingSize,
shape2d stride,
shape2d padding,
shape2d poolingSize,
shape2d stride,
shape2d padding,
ActivationType activationType
);
~AvgPooling2d();
@@ -36,9 +36,9 @@ class AvgPooling2d : public SequentialLayer, public TwoDLayer {
shape2d getOutputDims();
private:
protected:
shape2d inputSize;
int nChannels;
int nChannels;
shape2d poolingSize;
shape2d stride;
shape2d padding;
@@ -50,6 +50,11 @@ class AvgPooling2d : public SequentialLayer, public TwoDLayer {
Activation* activation;
};
class AdaptiveAvgPooling2d : public AvgPooling2d {
public:
AdaptiveAvgPooling2d(shape2d inputShape, int nChannels, shape2d outputShape, ActivationType activationType);
};
} // namespace CUDANet::Layers
#endif // CUDANET_AVG_POOLING_H