From 043fedde3c24cbd2f2053d9aa53ab2631094956d Mon Sep 17 00:00:00 2001 From: Iblis Lin Date: Wed, 21 Feb 2018 16:55:20 +0800 Subject: [PATCH] introduce Reexport - Reexporting NNlib fix #180 --- REQUIRE | 1 + src/Flux.jl | 6 ++---- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/REQUIRE b/REQUIRE index b75b59ca..8bb92ddb 100644 --- a/REQUIRE +++ b/REQUIRE @@ -9,6 +9,7 @@ GZip Colors ZipFile AbstractTrees +Reexport # AD ForwardDiff 0.5.0 diff --git a/src/Flux.jl b/src/Flux.jl index 939d6e2c..c1e18115 100644 --- a/src/Flux.jl +++ b/src/Flux.jl @@ -4,7 +4,7 @@ module Flux # Zero Flux Given -using Juno, Requires +using Juno, Requires, Reexport using MacroTools: @forward export Chain, Dense, RNN, LSTM, GRU, Conv2D, @@ -12,9 +12,7 @@ export Chain, Dense, RNN, LSTM, GRU, Conv2D, SGD, ADAM, Momentum, Nesterov, AMSGrad, param, params, mapleaves -using NNlib -export σ, sigmoid, logσ, logsigmoid, relu, leakyrelu, elu, swish, softmax, logsoftmax, - conv2d, maxpool2d, avgpool2d +@reexport using NNlib include("tracker/Tracker.jl") using .Tracker