Skip to content

Commit 529a482

Browse files
C_in -> channels_in
1 parent 06b6093 commit 529a482

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ext/NNlibCUDA/src/cudnn/pooling.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ 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)...), pdims.C_in)
45+
PoolDims{2,(1,K...),(1,S...),(0,0,P...),(1,D...)}((1,NNlib.input_size(pdims)...), NNlib.channels_in(pdims))
4646
end
4747

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

0 commit comments

Comments
 (0)