2017-06-05 18:12:40 +00:00
|
|
|
using Flux, DataFlow, MacroTools, Base.Test
|
2017-06-05 15:32:16 +00:00
|
|
|
using Flux: graph, Param, squeeze, unsqueeze, back!, update!, flatten
|
2017-02-21 15:46:38 +00:00
|
|
|
using DataFlow: Line, Frame
|
2016-12-15 23:11:35 +00:00
|
|
|
|
2017-02-01 07:11:17 +00:00
|
|
|
macro mxonly(ex)
|
|
|
|
|
:(Base.find_in_path("MXNet") ≠ nothing && $(esc(ex)))
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
macro tfonly(ex)
|
|
|
|
|
:(Base.find_in_path("TensorFlow") ≠ nothing && $(esc(ex)))
|
|
|
|
|
end
|
|
|
|
|
|
2017-02-20 19:34:57 +00:00
|
|
|
include("batching.jl")
|
2017-05-04 12:52:31 +00:00
|
|
|
include("backend/common.jl")
|
|
|
|
|
|
2016-12-15 21:05:53 +00:00
|
|
|
include("basic.jl")
|
2017-03-29 17:30:28 +00:00
|
|
|
include("recurrent.jl")
|
2017-06-26 09:21:17 +00:00
|
|
|
include("optimizer.jl")
|
|
|
|
|
|
2017-03-27 17:24:15 +00:00
|
|
|
@tfonly include("backend/tensorflow.jl")
|
2017-02-23 22:28:18 +00:00
|
|
|
@mxonly include("backend/mxnet.jl")
|
