load tf lazily to speed up load time

2016-12-16 15:54:15 +00:00 · 2016-12-16 15:54:15 +00:00 · ac374087b3
commit ac374087b3
parent c82716b535
1 changed files with 10 additions and 4 deletions
--- a/src/backend/backend.jl
+++ b/src/backend/backend.jl
@ -1,5 +1,11 @@
-# TODO: load backends lazily
-
-include("tensorflow/tensorflow.jl")
-using .TF
 export tf
+
+function loadtf()
+  isdefined(Flux, :TF) && return
+  @eval include(joinpath(dirname($@__FILE__), "tensorflow/tensorflow.jl"))
+end
+
+function tf(args...)
+  loadtf()
+  TF.tf(args...)
+end