Skip to content

Commit fca5b64

Browse files
committed
add tests for different kernel size, stride, pad and meanpool
1 parent 8b739fd commit fca5b64

File tree

2 files changed

+463
-25
lines changed

2 files changed

+463
-25
lines changed

src/impl/pooling_direct.jl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@ for name in (:max, :mean)
66
@eval function $((Symbol("$(name)pool_direct!")))(
77
y::AbstractArray{T,5}, x::AbstractArray{T,5},
88
pdims::PoolDims; alpha::T = T(1), beta::T = T(0)) where {T}
9+
@assert beta == T(0) "beta not supported yet"
910
check_dims(size(x), size(y), pdims)
1011

1112
width, height, depth = input_size(pdims)

0 commit comments

Comments
 (0)