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

conv_filter api changes

Browse Source
This commit is contained in:
Tejan Karmali 2018-10-24 13:28:22 -04:00
parent fca93471b3
commit 387df8c095
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/tracker/array.jl
View File

@ -359,7 +359,7 @@ conv(x::TrackedArray, w::AbstractArray; kw...) = track(conv, x, w; kw...)
conv(data(x), data(w); kw...),
Δ -> nobacksies(:conv,
(NNlib.∇conv_data(data.((Δ, w))...; size=size(x), kw...),
NNlib.∇conv_filter(data.((Δ, x, w))...; kw...)))
NNlib.∇conv_filter(data.((Δ, x))..., size(w); kw...)))
∇conv_data(x::TrackedArray, w::TrackedArray; kw...) = track(∇conv_data, x, w; kw...)
∇conv_data(x::AbstractArray, w::TrackedArray; kw...) = track(∇conv_data, x, w; kw...)
@ -369,7 +369,7 @@ conv(x::TrackedArray, w::AbstractArray; kw...) = track(conv, x, w; kw...)
∇conv_data(data(x), data(w); kw...),
Δ -> nobacksies(:conv,
(NNlib.conv(data.((Δ, w))...; size=size(x), kw...),
NNlib.∇conv_filter(data.((x, Δ, w))...; kw...)))
NNlib.∇conv_filter(data.((x, Δ))..., size(w); kw...)))
maxpool(x::TrackedArray, k; kw...) = track(maxpool, x, k; kw...)