From ca8ad63fb6021da6792ecc7b41d2f20ed933ac1c Mon Sep 17 00:00:00 2001 From: Tejan Karmali Date: Wed, 24 Oct 2018 12:52:43 -0400 Subject: [PATCH] in accordance with conv_data 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 e32efebf..beb7453e 100644 --- a/src/tracker/lib/array.jl +++ b/src/tracker/lib/array.jl @@ -383,7 +383,7 @@ conv(x::TrackedArray, w::AbstractArray; kw...) = track(conv, x, w; kw...) @grad conv(x, w; kw...) = conv(data(x), data(w); kw...), Δ -> nobacksies(:conv, - (NNlib.∇conv_data(data.((Δ, w))..., size(x); kw...), + (NNlib.∇conv_data(data.((Δ, w))...; size=size(x), kw...), NNlib.∇conv_filter(data.((Δ, x, w))...; kw...))) ∇conv_data(x::TrackedArray, w::TrackedArray; kw...) = track(∇conv_data, x, w; kw...) @@ -393,7 +393,7 @@ conv(x::TrackedArray, w::AbstractArray; kw...) = track(conv, x, w; kw...) @grad ∇conv_data(x, w; kw...) = ∇conv_data(data(x), data(w); kw...), Δ -> nobacksies(:conv, - (NNlib.conv(data.((Δ, w))..., size(x); kw...), + (NNlib.conv(data.((Δ, w))...; size=size(x), kw...), NNlib.∇conv_filter(data.((x, Δ, w))...; kw...))) maxpool(x::TrackedArray, k; kw...) = track(maxpool, x, k; kw...)