remove extra chainrules

Author: ChrisRackauckas

src/zygote.jl

@@ -1,68 +1,2 @@
-function ChainRulesCore.rrule(::typeof(getindex), VA::AbstractVectorOfArray,
-                              i::Union{Int, AbstractArray{Int}, CartesianIndex, Colon,
-                                       BitArray, AbstractArray{Bool}})
-    function AbstractVectorOfArray_getindex_adjoint(Δ)
-        Δ′ = [(i == j ? Δ : zero(x)) for (x, j) in zip(VA.u, 1:length(VA))]
-        (NoTangent(), VectorOfArray(Δ′), NoTangent())
-    end
-    VA[i], AbstractVectorOfArray_getindex_adjoint
-end
-
-function ChainRulesCore.rrule(::typeof(getindex), VA::AbstractVectorOfArray,
-                              indices::Union{Int, AbstractArray{Int}, CartesianIndex, Colon,
-                                             BitArray, AbstractArray{Bool}}...)
-    function AbstractVectorOfArray_getindex_adjoint(Δ)
-        Δ′ = zero(VA)
-        Δ′[indices...] = Δ
-        (NoTangent(), VectorOfArray(Δ′), map(_ -> NoTangent(), indices)...)
-    end
-    VA[indices...], AbstractVectorOfArray_getindex_adjoint
-end
-
-function ChainRulesCore.rrule(::Type{<:ArrayPartition}, x::S,
-                              ::Type{Val{copy_x}} = Val{false}) where {S <: Tuple, copy_x}
-    function ArrayPartition_adjoint(_y)
-        y = Array(_y)
-        starts = vcat(0, cumsum(reduce(vcat, length.(x))))
-        NoTangent(),
-        ntuple(i -> reshape(y[(starts[i] + 1):starts[i + 1]], size(x[i])), length(x)),
-        NoTangent()
-    end
-
-    ArrayPartition(x, Val{copy_x}), ArrayPartition_adjoint
-end
-
-function ChainRulesCore.rrule(::Type{<:VectorOfArray}, u)
-    VectorOfArray(u),
-    y -> (NoTangent(),
-          [y[ntuple(x -> Colon(), ndims(y) - 1)..., i] for i in 1:size(y)[end]])
-end
-
-function ChainRulesCore.rrule(::Type{<:DiffEqArray}, u, t)
-    DiffEqArray(u, t),
-    y -> (NoTangent(),
-          [y[ntuple(x -> Colon(), ndims(y) - 1)..., i] for i in 1:size(y)[end]],
-          NoTangent())
-end
-
-function ChainRulesCore.rrule(::typeof(getproperty), A::ArrayPartition, s::Symbol)
-    if s !== :x
-        error("$s is not a field of ArrayPartition")
-    end
-    function literal_ArrayPartition_x_adjoint(d)
-        (NoTangent(),
-         ArrayPartition((isnothing(d[i]) ? zero(A.x[i]) : d[i] for i in 1:length(d))...))
-    end
-    A.x, literal_ArrayPartition_x_adjoint
-end
-
 # Define a new species of projection operator for this type:
 ChainRulesCore.ProjectTo(x::VectorOfArray) = ChainRulesCore.ProjectTo{VectorOfArray}()
-
-# Gradient from iteration will be e.g. Vector{Vector}, this makes it another AbstractMatrix
-#(::ChainRulesCore.ProjectTo{VectorOfArray})(dx::AbstractVector{<:AbstractArray}) = VectorOfArray(dx)
-# Gradient from broadcasting will be another AbstractArray
-#(::ChainRulesCore.ProjectTo{VectorOfArray})(dx::AbstractArray) = dx
-
-# These rules duplicate the `rrule` methods above, because Zygote looks for an `@adjoint`
-# definition first, and finds its own before finding those.