more stack frame integration
This commit is contained in:
parent
353f156354
commit
3bfb2c5a37
@ -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
|
||||
|
@ -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
|
||||
|
||||
|
@ -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)
|
||||
|
Loading…
Reference in New Issue
Block a user