Flux.jl/src/backend/backend.jl

22 lines
383 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()
TF.tf(args...)
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()
MX.mxnet(args...)
end