Searched refs:jvp_subtrace (Results 1 – 1 of 1) sorted by relevance
43 return jvpfun(jvp_subtrace(fun), instantiate)63 def jvp_subtrace(main, primals, tangents): function297 f_jvp = jvp_subtrace(f, self.main)