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

unroll1

This commit is contained in:
Mike J Innes 2016-11-15 20:01:35 +00:00
parent 7f2ef71daf
commit 806febdd37
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/compiler/loops.jl
View File

@ -1,4 +1,4 @@
export unroll
export unroll, unroll1
type Offset
name::Symbol
@ -119,4 +119,10 @@ function unroll(model, n; seq = true, stateful = true)
seq || stateful ? Unrolled(model, graph, state, stateful, n) : Capacitor(graph)
end
function unroll1(model)
graph, state = unrollgraph(model, 1; seq = false)
graph = group(graph[1], map(x->x[1], inputs(graph)[2:end])...)
Unrolled(model, graph, state, false, 1)
end
flip(model) = Capacitor(map(x -> isa(x, Offset) ? -x : x, atomise(model)))