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

more appropriate naming

Browse Source
This commit is contained in:
Mike J Innes 2017-05-03 18:33:14 +01:00
parent 50245a16b0
commit 8c10a95bf4
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/compiler/code.jl
View File

@ -3,7 +3,7 @@ using MacroTools: @q
export @net export @net
function process_func(ex, params = []) function graphdef(ex, params = [])
@capture(shortdef(ex), (args__,) -> body_) @capture(shortdef(ex), (args__,) -> body_)
body = @> body MacroTools.flatten liftloops graphm DataFlow.il body = @> body MacroTools.flatten liftloops graphm DataFlow.il
body = mapconst(x -> x in params ? :(self.$x) : x, body) body = mapconst(x -> x in params ? :(self.$x) : x, body)
@ -67,7 +67,7 @@ function process_type(ex)
funcs = true || []] = groupby(x->isexpr(x, :->, :function), fs) funcs = true || []] = groupby(x->isexpr(x, :->, :function), fs)
@assert length(funcs) == 1 @assert length(funcs) == 1
pnames = namify.(params) pnames = namify.(params)
args, body = process_func(funcs[1], pnames) args, body = graphdef(funcs[1], pnames)
self = esc(:self) self = esc(:self)
quote quote
$(build_type(T, params)) $(build_type(T, params))
@ -79,7 +79,7 @@ function process_type(ex)
end end
function process_anon(ex) function process_anon(ex)
args, body = process_func(ex) args, body = graphdef(ex)
:(Capacitor($(DataFlow.constructor(mapconst(esc, makegraph(body, args)[1]))))) :(Capacitor($(DataFlow.constructor(mapconst(esc, makegraph(body, args)[1])))))
end end