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

add this for convenience

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

8
src/compiler/code.jl
View File

@ -83,10 +83,16 @@ function process_anon(ex)
:(Capacitor($(DataFlow.constructor(mapconst(esc, makegraph(body, args)[1]))))) :(Capacitor($(DataFlow.constructor(mapconst(esc, makegraph(body, args)[1])))))
end end
function process_def(ex)
# TODO: make a singleton net type
@capture(ex, f_(xs__) = body_)
:($(esc(f)) = @net $(esc(:(($(xs...),) -> $body))))
end
macro net(ex) macro net(ex)
ex = shortdef(ex) ex = shortdef(ex)
isexpr(ex, :type) ? process_type(ex) : isexpr(ex, :type) ? process_type(ex) :
@capture(ex, (__,) -> _) ? process_anon(ex) : @capture(ex, (__,) -> _) ? process_anon(ex) :
@capture(ex, _(__) = _) ? error("@net functions not implemented") : @capture(ex, _(__) = _) ? process_def(ex) :
error("Unsupported model expression $ex") error("Unsupported model expression $ex")
end end