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

graph op fixes

This commit is contained in:
Mike J Innes 2016-10-26 00:39:16 +01:00
parent eb78f67a93
commit ba60c4596b
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/compiler/graph.jl
View File

@ -2,16 +2,16 @@ immutable ModelInput end
inputnode(n) = vertex(Split(n), constant(ModelInput()))
# isinput(x) = isa(x, Constant) && value(x) == Input()
# function bumpinput(v::IVertex)
# prewalk(v) do v
# isa(value(v), Split) && value(v[1]) == Input() ?
# end
# end
function bumpinputs(v::IVertex)
prewalk(v) do v
isa(value(v), Split) && value(v[1]) == Constant(ModelInput()) ?
inputnode(value(v).n + 1) :
v
end
end
function spliceinput(v::IVertex, input::IVertex)
prewalk(v) do v
postwalk(v) do v
value(v) == Constant(ModelInput()) ? input : v
end
end
@ -20,7 +20,7 @@ spliceinputs(v::IVertex, inputs::Vertex...) =
spliceinput(v, vertex(Group(), inputs...))
function detuple(v::IVertex)
prewalk(v) do v
postwalk(v) do v
if isa(value(v), Split) && isa(value(v[1]), Group)
v[1][value(v).n]
else