From 704be49483bf746e75e3837fd9e55f5ad3d41ae4 Mon Sep 17 00:00:00 2001 From: Manjunath Bhat Date: Fri, 1 Mar 2019 15:04:05 +0530 Subject: [PATCH] Added support for Float64 for DepthwiseConv DepthwiseConv was giving errors for Float64. This fixes the issue. --- src/layers/conv.jl | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/layers/conv.jl b/src/layers/conv.jl index 216957a7..99a04890 100644 --- a/src/layers/conv.jl +++ b/src/layers/conv.jl @@ -165,6 +165,12 @@ function Base.show(io::IO, l::DepthwiseConv) print(io, ")") end +(a::DepthwiseConv{<:Any,<:Any,W})(x::AbstractArray{T}) where {T <: Union{Float32,Float64}, W <: AbstractArray{T}} = + invoke(a, Tuple{AbstractArray}, x) + +(a::DepthwiseConv{<:Any,<:Any,W})(x::AbstractArray{<:Real}) where {T <: Union{Float32,Float64}, W <: AbstractArray{T}} = + a(T.(x)) + """ MaxPool(k)