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

interpreter

Browse Source
This commit is contained in:
Mike J Innes 2016-11-15 16:58:59 +00:00
parent f31b539566
commit 7f2ef71daf
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/compiler/loops.jl
View File

@ -110,6 +110,8 @@ type Unrolled <: Model
steps::Int steps::Int
end end
(m::Unrolled)(xs...) = interpret(reifyparams(m.graph), xs...)
graph(u::Unrolled) = u.graph graph(u::Unrolled) = u.graph
function unroll(model, n; seq = true, stateful = true) function unroll(model, n; seq = true, stateful = true)

2
src/model.jl
View File

@ -48,6 +48,6 @@ type Capacitor <: Model
graph::IVertex{Any} graph::IVertex{Any}
end end
# TODO: Julia implementation that interprets the graph (m::Capacitor)(xs...) = interpret(reifyparams(m.graph), xs...)
graph(cap::Capacitor) = cap.graph graph(cap::Capacitor) = cap.graph