Skip to content

Commit 8e88632

Browse files
committed
Fix handling optional argument
1 parent 2181d0a commit 8e88632

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/nf_layer_constructors_submodule.f90

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -90,14 +90,14 @@ pure module function maxpool2d(pool_size, stride) result(res)
9090
stride_ = pool_size
9191
end if
9292

93-
if (stride < 1) &
93+
if (stride_ < 1) &
9494
error stop 'stride must be >= 1 in a maxpool2d layer'
9595

9696
res % name = 'maxpool2d'
9797

9898
allocate( &
9999
res % p, &
100-
source=maxpool2d_layer(pool_size, stride) &
100+
source=maxpool2d_layer(pool_size, stride_) &
101101
)
102102

103103
end function maxpool2d

0 commit comments

Comments
 (0)