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

more stack frame integration

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/Flux.jl
View File

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

5
src/backend/tensorflow/model.jl
View File

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

1
src/compiler/code.jl
View File

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