diff --git a/src/backend/tensorflow/graph.jl b/src/backend/tensorflow/graph.jl index e4cae4c0..c320c63a 100644 --- a/src/backend/tensorflow/graph.jl +++ b/src/backend/tensorflow/graph.jl @@ -19,7 +19,6 @@ 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.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) diff --git a/src/dims/seq.jl b/src/dims/seq.jl index ceadbbbe..d0080758 100644 --- a/src/dims/seq.jl +++ b/src/dims/seq.jl @@ -1,4 +1,4 @@ -export seq, Seq, BatchSeq +export Seq, BatchSeq struct Seq{T,S} <: AbstractVector{T} data::CatMat{T,S}