Flux.jl/src/backend/backend.jl

22 lines
393 B
Julia
Raw Normal View History

2017-01-28 17:02:49 +00:00
export tf, mxnet
2016-12-16 15:54:15 +00:00
function loadtf()
isdefined(Flux, :TF) && return
@eval include(joinpath(dirname($@__FILE__), "tensorflow/tensorflow.jl"))
end
function tf(args...)
loadtf()
2017-01-30 16:56:46 +00:00
eval(:(TF.tf($(args...))))
2016-12-16 15:54:15 +00:00
end
2017-01-28 17:02:49 +00:00
function loadmx()
isdefined(Flux, :MX) && return
@eval include(joinpath(dirname($@__FILE__), "mxnet/mxnet.jl"))
end
2017-03-08 21:41:13 +00:00
function mxnet(m)
2017-01-28 17:02:49 +00:00
loadmx()
2017-03-08 21:41:13 +00:00
eval(:(MX.mxnet($m)))
2017-01-28 17:02:49 +00:00
end