Skip to content

Commit 92dab93

Browse files
chore: add symmteric accum method for empty gradients (#1577)
1 parent a50c429 commit 92dab93

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/lib/lib.jl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ accum(x, y, zs...) = accum(accum(x, y), zs...)
1616
accum(x::Tuple, ys::Tuple...) = map(accum, x, ys...)
1717
accum(x::AbstractArray, ys::AbstractArray...) = Base.broadcast_preserving_zero_d(accum, x, ys...)
1818
accum(::Tuple{}, ::NamedTuple{}) = ()
19+
accum(::NamedTuple{}, ::Tuple{}) = ()
1920

2021
@generated function accum(x::NamedTuple, y::NamedTuple)
2122
# assumes that y has no keys apart from those also in x

0 commit comments

Comments
 (0)