@@ -3,7 +3,7 @@ using Statistics
3
3
# Pooling is so similar, we abstract over meanpooling and maxpooling, simply replacing
4
4
# the inner loop operation and a few initialization parameters.
5
5
for name in (:max , :mean )
6
- @eval @timeit_debug to function $ ((Symbol (" $(name) pool_direct!" )))(
6
+ @eval function $ ((Symbol (" $(name) pool_direct!" )))(
7
7
y:: AbstractArray{T,5} , x:: AbstractArray{T,5} ,
8
8
pdims:: PoolDims ; alpha:: T = T (1 ), beta:: T = T (0 )) where {T}
9
9
check_dims (size (x), size (y), pdims)
@@ -121,7 +121,7 @@ for name in (:max, :mean)
121
121
122
122
# Same story for gradients, and although this is very similar to the forward pass,
123
123
# it's unfortunately different enough that I think we need a separate function. :(
124
- @eval @timeit_debug to function $ ((Symbol (" ∇$(name) pool_direct!" )))(
124
+ @eval function $ ((Symbol (" ∇$(name) pool_direct!" )))(
125
125
dx:: AbstractArray{T,5} , dy:: AbstractArray{T,5} ,
126
126
y:: AbstractArray{T,5} , x:: AbstractArray{T,5} ,
127
127
pdims:: PoolDims ; alpha:: T = T (1 ), beta:: T = T (0 )) where {T}
0 commit comments