Skip to content

Commit 85947af

Browse files
fix constructor
1 parent 529a482 commit 85947af

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

ext/NNlibCUDA/src/cudnn/pooling.jl

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,12 @@ end
4242
add1d(x) = reshape(x, 1, size(x)...)
4343

4444
function fix_pooldims_1d(pdims::PoolDims{1,K,S,P,D}) where {K,S,P,D}
45-
PoolDims{2,(1,K...),(1,S...),(0,0,P...),(1,D...)}((1,NNlib.input_size(pdims)...), NNlib.channels_in(pdims))
45+
PoolDims{2, K + 1, S + 1, P + 2, D + 1}((1, NNlib.input_size(pdims)...),
46+
(1, K...),
47+
NNlib.channels_in(pdims),
48+
(1, S...),
49+
(0, 0, P...),
50+
(1, D...))
4651
end
4752

4853
function maxpool!(y::DenseCuArray{T,3}, x::DenseCuArray{T,3}, pdims::PoolDims) where T<:CUDNNFloat

0 commit comments

Comments
 (0)