From 73f345927a81720d12286f0b177f4e09cae0dbef Mon Sep 17 00:00:00 2001 From: Markus Hauru Date: Tue, 14 Jan 2025 17:29:29 +0000 Subject: [PATCH] Remove mapvcat Zygote manual adjoint --- ext/BijectorsZygoteExt.jl | 5 ----- 1 file changed, 5 deletions(-) diff --git a/ext/BijectorsZygoteExt.jl b/ext/BijectorsZygoteExt.jl index 79195b88..0db59007 100644 --- a/ext/BijectorsZygoteExt.jl +++ b/ext/BijectorsZygoteExt.jl @@ -22,7 +22,6 @@ using Bijectors: find_alpha, pd_logpdf_with_trans, istraining, - mapvcat, eachcolmaphcat, sumeachcol, pd_link, @@ -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)