Skip to content

Commit 4ef9495

Browse files
Update RecursiveArrayToolsZygoteExt.jl
1 parent d46cc44 commit 4ef9495

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

ext/RecursiveArrayToolsZygoteExt.jl

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ end
2020

2121
@adjoint function getindex(VA::AbstractVectorOfArray, i::Int)
2222
function AbstractVectorOfArray_getindex_adjoint(Δ)
23-
Δ′ = [(i == j ? Δ : Fill(zero(eltype(x)), size(x)))
23+
Δ′ = [(i == j ? Δ : FillArrays.Fill(zero(eltype(x)), size(x)))
2424
for (x, j) in zip(VA.u, 1:length(VA))]
2525
(VectorOfArray(Δ′), nothing)
2626
end
@@ -30,7 +30,7 @@ end
3030
@adjoint function getindex(VA::AbstractVectorOfArray,
3131
i::Union{BitArray, AbstractArray{Bool}})
3232
function AbstractVectorOfArray_getindex_adjoint(Δ)
33-
Δ′ = [(i[j] ? Δ[j] : Fill(zero(eltype(x)), size(x)))
33+
Δ′ = [(i[j] ? Δ[j] : FillArrays.Fill(zero(eltype(x)), size(x)))
3434
for (x, j) in zip(VA.u, 1:length(VA))]
3535
(VectorOfArray(Δ′), nothing)
3636
end
@@ -40,7 +40,7 @@ end
4040
@adjoint function getindex(VA::AbstractVectorOfArray, i::AbstractArray{Int})
4141
function AbstractVectorOfArray_getindex_adjoint(Δ)
4242
iter = 0
43-
Δ′ = [(j i ? Δ[iter += 1] : Fill(zero(eltype(x)), size(x)))
43+
Δ′ = [(j i ? Δ[iter += 1] : FillArrays.Fill(zero(eltype(x)), size(x)))
4444
for (x, j) in zip(VA.u, 1:length(VA))]
4545
(VectorOfArray(Δ′), nothing)
4646
end
@@ -50,7 +50,7 @@ end
5050
@adjoint function getindex(VA::AbstractVectorOfArray,
5151
i::Union{Int, AbstractArray{Int}})
5252
function AbstractVectorOfArray_getindex_adjoint(Δ)
53-
Δ′ = [(i[j] ? Δ[j] : Fill(zero(eltype(x)), size(x)))
53+
Δ′ = [(i[j] ? Δ[j] : FillArrays.Fill(zero(eltype(x)), size(x)))
5454
for (x, j) in zip(VA.u, 1:length(VA))]
5555
(VectorOfArray(Δ′), nothing)
5656
end

0 commit comments

Comments
 (0)