TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix missing parenthesis

Browse Source
This commit is contained in:
Avik Pal 2018-06-12 21:32:56 +05:30
parent f12e367cab
commit 24d13ac326
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/cuda/cudnn.jl
View File

@ -56,7 +56,7 @@ function cudnnBNForward(g::CuArray{T}, b::CuArray{T}, x::CuArray{T},
alpha = T(1), beta = T(0), alpha = T(1), beta = T(0),
eps = T(1e-5), training = true) where T<:Union{Float32, Float64} eps = T(1e-5), training = true) where T<:Union{Float32, Float64}
y = similar(x) y = similar(x)
cudnnBNForward!(y, g, b, x, running_mean, running_var, momentum, cache = cache cudnnBNForward!(y, g, b, x, running_mean, running_var, momentum, cache = cache,
alpha = alpha, beta = beta, eps = eps, training = training) alpha = alpha, beta = beta, eps = eps, training = training)
y y
end end
@ -125,7 +125,7 @@ end
function cudnnBNBackward!(dg::CuArray{T}, g::CuArray{T}, db::CuArray{T}, function cudnnBNBackward!(dg::CuArray{T}, g::CuArray{T}, db::CuArray{T},
dx::CuArray{T}, x::CuArray{T}, dy::CuArray{T}, dx::CuArray{T}, x::CuArray{T}, dy::CuArray{T},
running_mean::CuArray{T}, running_var::CuArray{T} running_mean::CuArray{T}, running_var::CuArray{T},
momentum; training = true, momentum; training = true,
cache = nothing, eps = T(1e-5), cache = nothing, eps = T(1e-5),
alpha = T(1), beta = T(0), alpha = T(1), beta = T(0),