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 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