From 05ac3855f4a7297b923203fdbcceb6aae6cc355f Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Mon, 27 Mar 2017 18:23:22 +0100 Subject: [PATCH] fixes --- src/backend/tensorflow/graph.jl | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/backend/tensorflow/graph.jl b/src/backend/tensorflow/graph.jl index 650a29e6..8f5b47bc 100644 --- a/src/backend/tensorflow/graph.jl +++ b/src/backend/tensorflow/graph.jl @@ -1,6 +1,7 @@ using Base: @get! using DataFlow: Constant, constant, Split using DataFlow.Interpreter +using DataFlow.Interpreter: stack using Flux: imap using TensorFlow: RawTensor, TFException @@ -19,7 +20,7 @@ graph(::typeof(σ), x) = nn.sigmoid(x) graph(::typeof(hcat), xs...) = concat(1, xs) graph(::typeof(seq), xs, n) = TensorFlow.unpack(xs, num = n, axis = 1) -for op in (tanh, *, .*, .+, .-) +for op in (tanh, *, .*, .+) @eval graph(::typeof($op), args...) = $op(args...) end