TastyPancakes
/
Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Give `NNPACK` a bit of numerical leeway

This commit is contained in:
Elliot Saba 2019-12-23 00:41:18 -08:00
parent 8a1e2f19d7
commit 0fdcc00923
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
test/layers/conv.jl
View File

@ -66,7 +66,7 @@ end
w = rand(2,2,1,1) w = rand(2,2,1,1)
y = CrossCor(w, [0.0]) y = CrossCor(w, [0.0])
@test sum(w .* x[1:2, 1:2, :, :]) == y(x)[1, 1, 1, 1] @test isapprox(sum(w .* x[1:2, 1:2, :, :]), y(x)[1, 1, 1, 1], rtol=1e-7)
r = zeros(Float32, 28, 28, 1, 5) r = zeros(Float32, 28, 28, 1, 5)
m = Chain( m = Chain(
@ -89,17 +89,17 @@ end
l = Conv((3,3), 1=>1) l = Conv((3,3), 1=>1)
expected = zeros(eltype(l.weight),5,5,1,1) expected = zeros(eltype(l.weight),5,5,1,1)
expected[2:end-1,2:end-1,1,1] = l.weight expected[2:end-1,2:end-1,1,1] = l.weight
@test expected == l(data) @test expected ≈ l(data)
l = Conv((3,1), 1=>1) l = Conv((3,1), 1=>1)
expected = zeros(eltype(l.weight),5,7,1,1) expected = zeros(eltype(l.weight),5,7,1,1)
expected[2:end-1,4,1,1] = l.weight expected[2:end-1,4,1,1] = l.weight
@test expected == l(data) @test expected ≈ l(data)
l = Conv((1,3), 1=>1) l = Conv((1,3), 1=>1)
expected = zeros(eltype(l.weight),7,5,1,1) expected = zeros(eltype(l.weight),7,5,1,1)
expected[4,2:end-1,1,1] = l.weight expected[4,2:end-1,1,1] = l.weight
@test expected == l(data) @test expected ≈ l(data)
@test begin @test begin
# we test that the next expression does not throw # we test that the next expression does not throw