Flux.jl/src/backend/backend.jl

22 lines
407 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
function mxnet(args...)
loadmx()
2017-01-30 16:56:46 +00:00
eval(:(MX.mxnet($(args...))))
2017-01-28 17:02:49 +00:00
end