diff --git a/src/backend/tensorflow/graph.jl b/src/backend/tensorflow/graph.jl index 61ebe9c4..28965646 100644 --- a/src/backend/tensorflow/graph.jl +++ b/src/backend/tensorflow/graph.jl @@ -20,7 +20,7 @@ graph(::typeof(softmax), x) = nn.softmax(x) graph(::typeof(relu), x) = nn.relu(x) 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) +graph(::typeof(seq), xs, n) = TensorFlow.unstack(xs, num = n, axis = 1) graph(::typeof(sum), x, dim=nothing) = TensorFlow.reduce_sum(x;axis=dim) graph(::typeof(prod), x, dim=nothing) = TensorFlow.reduce_prod(x;axis=dim) graph(::typeof(min), x, dim=nothing) = TensorFlow.reduce_min(x;axis=dim)