diff --git a/src/backend/mxnet/graph.jl b/src/backend/mxnet/graph.jl index 8fb5e6c1..d9f79c94 100644 --- a/src/backend/mxnet/graph.jl +++ b/src/backend/mxnet/graph.jl @@ -69,7 +69,7 @@ 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) = p.value function graph(ctx::Context, model, args...) g = Flux.graph(model) diff --git a/src/backend/tensorflow/graph.jl b/src/backend/tensorflow/graph.jl index 57dfa7c9..730c42e4 100644 --- a/src/backend/tensorflow/graph.jl +++ b/src/backend/tensorflow/graph.jl @@ -51,7 +51,7 @@ interp{T<:AArray}(ctx, p::Constant{Flux.Param{T}}) = ctx[:params][p.value] : (ctx[:params][p.value] = Variable(convertel(Float32, p.value.x))) -interp(ctx, p::Constant) = node(p.value) +interp(ctx, p::Constant) = p.value function interp(ctx, model, args...) g = Flux.graph(model)