This commit is contained in:
Mike J Innes 2016-11-13 18:16:58 +00:00
parent e4a6ca5f9e
commit 6ac4dd8429
2 changed files with 7 additions and 7 deletions

View File

@ -35,13 +35,6 @@ graph(::DataFlow.Group, xs...) = (xs...,)
graph(params::Associative, c::Conv2D, x) =
nn.conv2d(x, graph(params, c.filter), [1,c.stride...,1], "VALID")
type Op
f
shape
end
Op(f) = Op(f, (d...) -> nothing)
graph(op::Op, xs...) = op.f(xs...)
Flux.shape(op::Op, d...) = op.shape(d...)

View File

@ -5,6 +5,13 @@ import Flux: accuracy, spliceinputs, detuple
export tf
type Op
f
shape
end
Op(f) = Op(f, (d...) -> nothing)
include("graph.jl")
include("model.jl")
include("recurrent.jl")