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
|