2016-09-27 01:16:49 +00:00
|
|
|
module TF
|
|
|
|
|
|
2016-10-31 12:38:18 +00:00
|
|
|
using ..Flux, DataFlow, TensorFlow, Juno
|
2017-01-25 12:40:56 +00:00
|
|
|
import Flux: accuracy, rebatch, convertel
|
2016-09-29 19:50:43 +00:00
|
|
|
|
|
|
|
|
export tf
|
2016-09-27 01:16:49 +00:00
|
|
|
|
2017-03-14 17:56:03 +00:00
|
|
|
struct Op
|
2016-11-13 18:16:58 +00:00
|
|
|
f
|
|
|
|
|
shape
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
Op(f) = Op(f, (d...) -> nothing)
|
|
|
|
|
|
2016-11-13 20:27:20 +00:00
|
|
|
Flux.shape(op::Op, d...) = op.shape(d...)
|
|
|
|
|
|
2016-10-26 13:25:10 +00:00
|
|
|
include("graph.jl")
|
|
|
|
|
include("model.jl")
|
2016-10-10 22:04:26 +00:00
|
|
|
|
2016-09-27 01:16:49 +00:00
|
|
|
end
|
