Searched refs:unbound_vjp (Results 1 – 1 of 1) sorted by relevance
118 def unbound_vjp(pvals, jaxpr, consts, *cts): function126 vjp_ = Partial(partial(unbound_vjp, pvals, jaxpr), consts)