diff --git a/test/basic.jl b/test/basic.jl index e18cd37b..de9acaa8 100644 --- a/test/basic.jl +++ b/test/basic.jl @@ -1,21 +1,3 @@ -@net type TLP - first - second - function (x) - l1 = σ(first(x)) - l2 = softmax(second(l1)) - end -end - -@net type Multi - W - V - (x, y) -> (x*W, y*V) -end - -Multi(in::Integer, out::Integer) = - Multi(randn(in, out), randn(in, out)) - @testset "Basics" begin xs = randn(10) diff --git a/test/runtests.jl b/test/runtests.jl index 53be169a..97bb9e70 100644 --- a/test/runtests.jl +++ b/test/runtests.jl @@ -13,6 +13,24 @@ macro tfonly(ex) :(Base.find_in_path("TensorFlow") ≠ nothing && $(esc(ex))) end +@net type TLP + first + second + function (x) + l1 = σ(first(x)) + l2 = softmax(second(l1)) + end +end + +@net type Multi + W + V + (x, y) -> (x*W, y*V) +end + +Multi(in::Integer, out::Integer) = + Multi(randn(in, out), randn(in, out)) + include("batching.jl") include("basic.jl") include("recurrent.jl")