21 lines
255 B
Julia
21 lines
255 B
Julia
module TF
|
|
|
|
using ..Flux, DataFlow, TensorFlow, Juno
|
|
import Flux: accuracy, rebatch, convertel
|
|
|
|
export tf
|
|
|
|
struct Op
|
|
f
|
|
shape
|
|
end
|
|
|
|
Op(f) = Op(f, (d...) -> nothing)
|
|
|
|
Flux.shape(op::Op, d...) = op.shape(d...)
|
|
|
|
include("graph.jl")
|
|
include("model.jl")
|
|
|
|
end
|