in accordance with conv_filter api
This commit is contained in:
parent
10f3a8eae2
commit
bc9bda9a85
@ -384,7 +384,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))..., size(w); kw...)))
|
||||
NNlib.∇conv_filter(data.((Δ, x))...; size=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...)
|
||||
@ -394,7 +394,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, Δ))..., size(w); kw...)))
|
||||
NNlib.∇conv_filter(data.((x, Δ))...; size=size(w), kw...)))
|
||||
|
||||
maxpool(x::TrackedArray, k; kw...) = track(maxpool, x, k; kw...)
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user