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

a little shorter

Browse Source
This commit is contained in:
Mike J Innes 2016-06-03 00:57:52 +01:00
parent 8b7d575efe
commit dfc75df407
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/rt/diff.jl
View File

@ -24,10 +24,7 @@ function ∇graph(v::IVertex, ∇, out = d())
end end
macro derive(ex) macro derive(ex)
v = vertex(Flow.Do()) ∇s = ∇graph(il(graphm(resolve_calls(ex))), @flow(∇))
for (k, x) in ∇graph(il(graphm(resolve_calls(ex))), @flow(∇)) v = vertex(Flow.Do(), (@v(Flow.Assign(Symbol("∇", k))(v)) for (k, v) in ∇s)...)
k = Symbol("∇", k)
thread!(v, @v(Flow.Assign(k)(x)))
end
Expr(:quote, @> v cse syntax prettify) Expr(:quote, @> v cse syntax prettify)
end end