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

make interpmap composable

This commit is contained in:
Mike J Innes 2016-11-17 11:26:58 +00:00
parent a6169ec2d0
commit 3e42db0cdf
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/compiler/interp.jl
View File

@ -11,11 +11,15 @@ function astuples(xs)
all(x->!(x==nothing), xs) ? xs : nothing
end
function interp(ctx, ::typeof(map), f, xs...)
f = interpret(ctx, f)
xs = interpret(ctx, xs)
xs′ = astuples(xs)
xs′ == nothing ? vertex(map, constant(f), xs...) : group(map(f, xs′...)...)
function interpmap(cb)
function interp(ctx, ::typeof(map), f, xs...)
f, xs = interpret(ctx, (f, xs))
xs′ = astuples(xs)
xs′ ≠ nothing ?
group(map(f, xs′...)...) :
cb(ctx, map, constant(f), xs...)
end
interp(args...) = cb(args...)
end
function interp(ctx, model, xs...)
@ -25,4 +29,4 @@ function interp(ctx, model, xs...)
end
expand(graph, xs...) =
interp(Context(interplambda(interpconst(interptuple(interp)))), graph, xs...)
interp(Context(interplambda(interpmap(interpconst(interptuple(interp))))), graph, xs...)