Skip to content

Commit

Permalink
Remove mapvcat Zygote manual adjoint
Browse files Browse the repository at this point in the history
  • Loading branch information
mhauru committed Jan 14, 2025
1 parent c3fe929 commit 73f3459
Showing 1 changed file with 0 additions and 5 deletions.
5 changes: 0 additions & 5 deletions ext/BijectorsZygoteExt.jl
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,6 @@ using Bijectors:
find_alpha,
pd_logpdf_with_trans,
istraining,
mapvcat,
eachcolmaphcat,
sumeachcol,
pd_link,
Expand All @@ -36,10 +35,6 @@ using Bijectors.Distributions: LocationScale

@adjoint istraining() = true, _ -> nothing

@adjoint function mapvcat(f, args...)
g(f, args...) = map(f, args...)
return pullback(g, f, args...)
end
@adjoint function eachcolmaphcat(f, x1, x2)
function g(f, x1, x2)
init = reshape(f(view(x1, :, 1), x2[1]), :, 1)
Expand Down

0 comments on commit 73f3459

Please sign in to comment.