From 77bb2a66de4f514da39dc4e0e20f1c8dacd9b332 Mon Sep 17 00:00:00 2001 From: Keno Fischer Date: Tue, 23 Oct 2018 16:15:39 -0400 Subject: [PATCH] Use lower level conv interface --- src/layers/conv.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/layers/conv.jl b/src/layers/conv.jl index e3c6ae67..edd63075 100644 --- a/src/layers/conv.jl +++ b/src/layers/conv.jl @@ -41,7 +41,7 @@ function (c::Conv{<:Any, <:Any, <:Any, stride, pad, dilation})(x) where {stride, # TODO: breaks gpu broadcast :( # ndims(x) == ndims(c.weight)-1 && return squeezebatch(c(reshape(x, size(x)..., 1))) σ, b = c.σ, reshape(c.bias, map(_->1, stride)..., :, 1) - σ.(conv(x, c.weight, stride = stride, pad = pad, dilation = dilation) .+ b) + σ.(NNlib._conv{pad, stride, dilation}()(x, c.weight) .+ b) end function Base.show(io::IO, l::Conv)