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

factor out var creation

Browse Source
This commit is contained in:
Mike J Innes 2017-02-21 08:52:56 +00:00
parent 12cde694b3
commit 4fb51d1752
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/backend/mxnet/graph.jl
View File

@ -57,12 +57,14 @@ end
register(ctx::Context, node) = node register(ctx::Context, node) = node
function graph{T<:AArray}(ctx::Context, p::Constant{Flux.Param{T}}) function var(ctx::Context, p::Flux.Param)
id = gensym() id = gensym()
ctx[:params][id] = p.value.x ctx[:params][id] = p.x
return mx.Variable(id) return mx.Variable(id)
end end
graph{T<:AArray}(ctx::Context, p::Constant{Flux.Param{T}}) = var(ctx, p.value)
graph(ctx::Context, p::Constant) = node(p.value) graph(ctx::Context, p::Constant) = node(p.value)
function graph(ctx::Context, model, args...) function graph(ctx::Context, model, args...)