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

fix self escaping

Browse Source
This commit is contained in:
Mike J Innes 2016-12-21 13:04:54 +00:00
parent c720ba94d0
commit 6acfcd913e
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/compiler/code.jl
View File

@ -50,10 +50,11 @@ function process_type(ex)
pnames = namify.(params)
args, body = process_func(funcs[1], pnames)
@assert length(args) == 1
self = esc(:self)
quote
$(build_type(T, params))
$(esc(:(self::$T)))($(args...),) = interpret(reifyparams(graph(self)), $(args...))
$(esc(:(Flux.update!(self::$T, η)))) = ($(map(p -> :(update!(self.$p, η)), pnames)...);)
$(esc(:(self::$T)))($(args...),) = interpret(reifyparams(graph($self)), $(args...))
$(esc(:(Flux.update!(self::$T, η)))) = ($(map(p -> :(update!($self.$p, η)), pnames)...);)
$(esc(:(Flux.graph(self::$T)))) = $(DataFlow.constructor(mapconst(esc, makegraph(body, args))))
nothing
end