Searched refs:make_vjp (Results 1 – 2 of 2) sorted by relevance
504 val, vjp = make_vjp(f, params)(*args, **kwds)510 def make_vjp(f, params=None, persistent=True): function
564 wrapped_fn = backprop.make_vjp(f, persistent=False)574 wrapped_fn = backprop.make_vjp(f, persistent=True)