Searched refs:fun_jvp_partial_eval (Results 1 – 1 of 1) sorted by relevance
712 def fun_jvp_partial_eval(trace, *tracers, **params): function728 pe.custom_partial_eval_rules[fun_jvp_p] = fun_jvp_partial_eval