TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

basic unrolling

Browse Source
This commit is contained in:
Mike J Innes 2016-08-29 16:23:43 +01:00
parent 19b5e8bd21
commit cd968af228
1 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/compiler/loops.jl
View File

@ -20,9 +20,22 @@ bumpinput(x) = x
bumpinputs(v::IVertex) = mapconst(bumpinput, v) bumpinputs(v::IVertex) = mapconst(bumpinput, v)
function unroll(delay::IVertex)
prewalk(delay[1]) do v
isa(value(v), Delay) ? constant(ModelInput(1)) : v
end
end
function break!(model) function break!(model)
iscyclic(graph(model)) || return model iscyclic(graph(model)) || return model
bumpinputs(graph(model)) g = bumpinputs(graph(model))
loops = []
g = prewalk(g) do v
isa(value(v), Delay) || return v
push!(loops, unroll(v))
constant(ModelInput(1))
end
cse(vertex(tuple, loops..., g))
end end
# r = Recurrent(784, 10, 50) # r = Recurrent(784, 10, 50)