diff --git a/src/backend/mxnet/graph.jl b/src/backend/mxnet/graph.jl index c6f17e1e..2e406dba 100644 --- a/src/backend/mxnet/graph.jl +++ b/src/backend/mxnet/graph.jl @@ -32,6 +32,7 @@ graph(::typeof(broadcast), ::typeof(+), args...) = mx.broadcast_plus(args...) graph(::typeof(broadcast), ::typeof(*), args...) = mx.broadcast_mul(args...) graph(::typeof(broadcast), ::typeof(-), args...) = mx.broadcast_sub(args...) # Old broadcasters +graph(::typeof(broadcast), ::typeof(exp), xs) = exp(xs) graph(::typeof(.+), args...) = mx.broadcast_plus(args...) graph(::typeof(.*), args...) = mx.broadcast_mul(args...) graph(::typeof(.-), args...) = mx.broadcast_sub(args...)