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

Add padding to avg pooling

Browse Source
This commit is contained in:
Matúš Námešný
2024-05-26 18:54:12 +02:00
parent d1fc45d9e0
commit 4a67b708f0
6 changed files with 60 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
include/kernels/pooling.cuh
View File

@@ -23,7 +23,8 @@ __global__ void avg_pooling(
const dim2d outputSize,
const int nChannels,
const dim2d poolingSize,
const dim2d stride
const dim2d stride,
const dim2d padding
);
} // namespace CUDANet::Kernels

2
include/layers/avg_pooling.cuh
View File

@@ -13,6 +13,7 @@ class AvgPooling2d : public SequentialLayer, public TwoDLayer {
int nChannels,
dim2d poolingSize,
dim2d stride,
dim2d padding,
ActivationType activationType
);
~AvgPooling2d();
@@ -40,6 +41,7 @@ class AvgPooling2d : public SequentialLayer, public TwoDLayer {
int nChannels;
dim2d poolingSize;
dim2d stride;
dim2d padding;
dim2d outputSize;