Added Float64 tests for DepthwiseConv

2019-03-09 20:17:22 +05:30 · 2019-03-09 20:17:22 +05:30 · d4a1d33a31
parent 704be49483
commit d4a1d33a31
1 changed files with 10 additions and 0 deletions
--- a/test/layers/conv.jl
+++ b/test/layers/conv.jl
@ -32,4 +32,14 @@ end
  m2 = DepthwiseConv((2, 2), 3)

  @test size(m2(r), 3) == 3
+  
+  x = zeros(Float64, 28, 28, 3, 5)
+  
+  m3 = DepthwiseConv((2, 2), 3 => 5)
+  
+  @test size(m3(r), 3) == 15
+  
+  m4 = DepthwiseConv((2, 2), 3)
+  
+  @test size(m4(r), 3) == 3
 end