Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-12-23 06:44:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Migrate batch norm layer

Browse Source
This commit is contained in:
Matúš Námešný
2025-11-21 23:24:14 +01:00
parent 5679dc0a50
commit fd4775faa4
11 changed files with 181 additions and 364 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/layers/avg_pooling.cpp
View File

@@ -51,7 +51,7 @@ AvgPool2d::~AvgPool2d() {}
CUDANet::Tensor& AvgPool2d::forward(CUDANet::Tensor& input) {
output.zero();
backend->avgPool2d(
backend->avg_pool2d(
input,
output,
in_shape,