Skip to content

Commit b3ff0c2

Browse files
author
Alex Robson
committed
Fix up pullback
1 parent 64e7780 commit b3ff0c2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/chainrules.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
function ChainRulesCore.rrule(::typeof(*), A::Real, B::WoodburyPDMat)
44
project_A = ProjectTo(A)
55
project_B = ProjectTo(B)
6-
function times_pullback(::AbstractMatrix)
6+
function times_pullback(Ȳ::AbstractMatrix)
77
Ā = @thunk(project_A(dot(Ȳ, B)'))
88
= @thunk(project_B(A' * Ȳ))
99
return (NoTangent(), Ā, B̄)

0 commit comments

Comments
 (0)