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 activation to pooling layers

Browse Source
This commit is contained in:
Matúš Námešný
2024-04-11 19:50:54 +02:00
parent e86e04f6d6
commit bc86ed1782
2 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/layers/avg_pooling.cu
View File

@@ -43,5 +43,7 @@ float* AvgPooling2D::forward(const float* d_input) {
d_input, d_output, inputSize, outputSize, nChannels, poolingSize, stride
);
activation.activate(d_output);
return d_output;
}

2
src/layers/max_pooling.cu
View File

@@ -46,5 +46,7 @@ float* MaxPooling2D::forward(const float* d_input) {
d_input, d_output, inputSize, outputSize, nChannels, poolingSize, stride
);
activation.activate(d_output);
return d_output;
}