diff --git a/src/backend/mxnet/model.jl b/src/backend/mxnet/model.jl index 2488e3c1..3ea9ea12 100644 --- a/src/backend/mxnet/model.jl +++ b/src/backend/mxnet/model.jl @@ -97,6 +97,14 @@ end mxnet(model, ctx = :cpu) = Model(model, toctx(ctx)) +function Base.show(io::IO, m::Model) + print(io, "MX.Model(") + show(io, m.model) + print(io, ", ") + show(io, m.ctx) + print(io, ")") +end + import Base: @get! # TODO: dims having its own type would be useful diff --git a/src/compiler/interp.jl b/src/compiler/interp.jl index 49d78946..85618127 100644 --- a/src/compiler/interp.jl +++ b/src/compiler/interp.jl @@ -35,3 +35,5 @@ end (m::Capacitor)(xs...) = interpmodel(m, xs...) graph(cap::Capacitor) = cap.graph + +Base.show(io::IO, ::Capacitor) = print(io, "Capacitor(...)") diff --git a/src/compiler/loops.jl b/src/compiler/loops.jl index 72b882f3..0df64225 100644 --- a/src/compiler/loops.jl +++ b/src/compiler/loops.jl @@ -9,6 +9,12 @@ end Stateful(model, ss) = Stateful(model, ss, state.(ss), state.(ss)) +function Base.show(io::IO, m::Stateful) + print(io, "Stateful(") + show(io, m.model) + print(io, ")") +end + function (m::Stateful)(xs...) m.istate = m.ostate state, y = m.model((m.istate...,), xs...)