Add padding to max pooling

2025-11-05 17:34:21 +00:00 · 2024-05-26 19:03:10 +02:00
parent 4a67b708f0
commit 94a16b4352
7 changed files with 78 additions and 19 deletions
--- a/include/kernels/pooling.cuh
+++ b/include/kernels/pooling.cuh
@@ -13,7 +13,8 @@ __global__ void max_pooling(
    const dim2d outputSize,
    const int nChannels,
    const dim2d poolingSize,
-    const dim2d stride
+    const dim2d stride,
+    const dim2d padding
 );

 __global__ void avg_pooling(