From bc9bda9a85790c0cbe0a3db371ff3cc486319f4f Mon Sep 17 00:00:00 2001 From: Tejan Karmali Date: Fri, 26 Oct 2018 11:31:34 -0400 Subject: [PATCH] in accordance with conv_filter api --- src/tracker/lib/array.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/tracker/lib/array.jl b/src/tracker/lib/array.jl index f2e73712..4abd6e03 100644 --- a/src/tracker/lib/array.jl +++ b/src/tracker/lib/array.jl @@ -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...)