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

Implement avg pooling

Browse Source
This commit is contained in:
Matúš Námešný
2024-03-19 22:33:43 +01:00
parent a0fc1b00ae
commit ef63cbd9f1
6 changed files with 92 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
include/kernels/pooling.cuh
View File

@@ -11,8 +11,7 @@ __global__ void max_pooling(
const int inputSize,
const int nChannels,
const int poolingSize,
const int stride,
const int paddingSize
const int stride
);
__global__ void avg_pooling(
@@ -21,8 +20,7 @@ __global__ void avg_pooling(
const int inputSize,
const int nChannels,
const int poolingSize,
const int stride,
const int paddingSize
const int stride
);
} // namespace CUDANet::Kernels