From 8bf5d916057c7c9df609406565a2ac24b4614cf8 Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Sat, 28 Jan 2017 23:36:58 +0530 Subject: [PATCH] symbolname utility --- src/backend/mxnet/graph.jl | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/src/backend/mxnet/graph.jl b/src/backend/mxnet/graph.jl index 6a7db27b..0e6b9aa4 100644 --- a/src/backend/mxnet/graph.jl +++ b/src/backend/mxnet/graph.jl @@ -1,3 +1,11 @@ +function symbolname(s::mx.SymbolicNode) + name = Ref{mx.char_p}(0) + success = Ref(0) + mx.@mxcall(:MXSymbolGetName, (mx.MX_handle, Ref{mx.char_p}, Ref{Int}), s.handle.value, name, success) + @assert success[] != -1 + return Symbol(unsafe_wrap(String, name[])) +end + using Base: @get! using DataFlow: Constant, constant, Context, interpret, Split, interpv, ituple, ilambda, iconst, iline, stack, mux