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

use ituple properly

This commit is contained in:
Mike J Innes 2017-01-30 22:55:12 +05:30
parent a73b53e05e
commit 508cd730c9
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/backend/mxnet/graph.jl
View File

@ -18,7 +18,6 @@ node(x::mx.SymbolicNode) = x
# node(x::Number) = TensorFlow.constant(Float32(x))
graph(::typeof(tuple), args...) = (args...,)
graph(s::Split, t::Tuple) = t[s.n]
graph(::typeof(*), args...) = mx.dot(args...)
graph(::typeof(+), args...) = mx.broadcast_plus(args...)
graph(::typeof(σ), x) = mx.Activation(data = x, act_type = :sigmoid)
@ -74,7 +73,7 @@ function interp(ctx, model, args...)
end
function tograph(model, args...)
ctx = Context(mux(iline, ilambda, imap, iargs, interp),
ctx = Context(mux(iline, ilambda, imap, iargs, ituple, interp),
params = Dict(), stacks = Dict())
out = interp(ctx, model, map(constant, args)...)
return ctx[:params], ctx[:stacks], out