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
|