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

quote was the right thing here

Browse Source
This commit is contained in:
Mike J Innes 2017-02-24 15:48:52 +00:00
parent 5ac09338ba
commit f3d0c21952
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/compiler/code.jl
View File

@ -1,5 +1,4 @@
import DataFlow: mapconst, cse import DataFlow: mapconst, cse
import MacroTools: @q
export @net, @ml export @net, @ml
@ -69,7 +68,7 @@ function process_type(ex)
args, body = process_func(funcs[1], pnames) args, body = process_func(funcs[1], pnames)
@assert length(args) == 1 @assert length(args) == 1
self = esc(:self) self = esc(:self)
@q begin quote
$(build_type(T, params)) $(build_type(T, params))
$(esc(:(Flux.runmodel(self::$T, $(args...)) = $(build_forward(body, args))))) $(esc(:(Flux.runmodel(self::$T, $(args...)) = $(build_forward(body, args)))))
($self::$(esc(T)))($(map(arg -> :($arg::Batch), args)...)) = rebatch(runmodel($self, $(map(x->:(rawbatch($x)), args)...))) ($self::$(esc(T)))($(map(arg -> :($arg::Batch), args)...)) = rebatch(runmodel($self, $(map(x->:(rawbatch($x)), args)...)))