File tree Expand file tree Collapse file tree
DifferentiationInterface/ext/DifferentiationInterfaceZygoteExt Expand file tree Collapse file tree Original file line number Diff line number Diff line change 159159
160160function DI. hvp (
161161 f,
162- prep:: DI.ForwardOverReverseHVPPrep ,
162+ prep:: DI.ForwardOverAnythingHVPPrep ,
163163 backend:: AutoZygote ,
164164 x,
165165 tx:: NTuple ,
171171function DI. hvp! (
172172 f,
173173 tg:: NTuple ,
174- prep:: DI.ForwardOverReverseHVPPrep ,
174+ prep:: DI.ForwardOverAnythingHVPPrep ,
175175 backend:: AutoZygote ,
176176 x,
177177 tx:: NTuple ,
184184
185185function DI. gradient_and_hvp (
186186 f,
187- prep:: DI.ForwardOverReverseHVPPrep ,
187+ prep:: DI.ForwardOverAnythingHVPPrep ,
188188 backend:: AutoZygote ,
189189 x,
190190 tx:: NTuple ,
@@ -199,7 +199,7 @@ function DI.gradient_and_hvp!(
199199 f,
200200 grad,
201201 tg:: NTuple ,
202- prep:: DI.ForwardOverReverseHVPPrep ,
202+ prep:: DI.ForwardOverAnythingHVPPrep ,
203203 backend:: AutoZygote ,
204204 x,
205205 tx:: NTuple ,
You can’t perform that action at this time.
0 commit comments