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

make interpmap composable

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