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

shapes macro

Browse Source
This commit is contained in:
Mike J Innes 2017-02-01 21:06:02 +05:30
parent cd86dfdf07
commit a041da3fdc
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/compiler/shape.jl
View File

@ -1,5 +1,7 @@
using DataFlow.Interpreter using DataFlow.Interpreter
export @shapes
type Hint type Hint
typ typ
end end
@ -46,3 +48,13 @@ end
# TODO: make correct # TODO: make correct
infer(::typeof(+), a, b) = a infer(::typeof(+), a, b) = a
# Shapes macro
_shape(xs::AbstractArray) = size(xs)
_shape(xs::Tuple) = map(_shape, xs)
macro shapes(ex)
@capture(ex, f_(args__)) || error("@shapes f(args...)")
:(shapes($(esc(f)), _shape(($(map(esc, args)...),))...))
end