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

split out delay node grabber

Browse Source
This commit is contained in:
Mike J Innes 2016-06-07 17:00:09 +01:00
parent 823dbf621f
commit 5c018bcae7
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/compiler/loop.jl
View File

@ -1,9 +1,14 @@
function cut_forward(v::IVertex) function delays(v::IVertex)
pushes = [] ds = []
Flow.prefor(v) do w Flow.prefor(v) do w
value(w) == :Delay && value(w) == :Delay &&
push!(pushes, vertex(:push!, vertex(:(self.delay)), w[1])) push!(ds, w)
end end
return ds
end
function cut_forward(v::IVertex)
pushes = map(x->vertex(:push!, vertex(:(self.delay)), v[1]), delays(v))
isempty(pushes) && return v isempty(pushes) && return v
@assert length(pushes) == 1 @assert length(pushes) == 1
v = vertex(Flow.Do(), pushes..., v) v = vertex(Flow.Do(), pushes..., v)