more stack frame integration

This commit is contained in:
Mike J Innes 2016-12-26 12:11:24 +00:00
parent 353f156354
commit 3bfb2c5a37
3 changed files with 5 additions and 3 deletions

View File

@ -4,7 +4,7 @@ using MacroTools, Lazy, DataFlow, Juno
using DataFlow: graphm, syntax, prewalk!, postwalk!, prewalk, postwalk,
iscyclic, Constant, constant, isconstant, group, Split, splitnode,
detuple, value, inputs, thread!, value, inputs, Split, splitnode, inputnode,
spliceinputs, bumpinputs
spliceinputs, bumpinputs, Frame
using Juno: Tree, Row
# Zero Flux Given

View File

@ -41,8 +41,9 @@ function tferr(model::Model, e)
m == nothing && return
node = m.captures[1]
if haskey(model.stacks, node)
l = model.stacks[node][end]
println("TensorFlow error occured at $(l.file):$(l.line)")
stk = model.stacks[node]
println("TensorFlow error occured at:")
foreach(l -> println("$(l.file):$(l.line)"), stk)
end
end

View File

@ -21,6 +21,7 @@ function makegraph(graph, args)
:(Flux.Offset($(Expr(:quote, x.name)), $(x.n), self.$(x.name))) :
x
end
vertex(:(Flux.Frame(self)), graph)
end
function build_type(T, params)