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

avoid line numbers in runmodel

Browse Source
This commit is contained in:
Mike J Innes 2017-03-09 01:10:44 +00:00
parent 5c548fe1c4
commit 3f919aa81e
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/compiler/code.jl
View File

@ -1,4 +1,5 @@
import DataFlow: mapconst, cse
using MacroTools: @q
export @net, @ml
@ -72,7 +73,7 @@ function process_type(ex)
self = esc(:self)
quote
$(build_type(T, params))
$(esc(:(Flux.runmodel(self::$T, $(args...)) = $(build_forward(body, args)))))
$(@q $(esc(:(Flux.runmodel(self::$T, $(args...)) = $(build_forward(body, args))))))
($self::$(esc(T)))($(args...)) = runrawbatched((xs...) -> runmodel($self, xs...), $(args...))
$(esc(:(Flux.update!(self::$T, η)))) = ($(map(p -> :(update!($self.$p, η)), pnames)...);)
$(esc(:(Flux.graph(self::$T)))) = $(DataFlow.constructor(mapconst(esc, makegraph(body, args))))