Searched refs:vjpmaker (Results 1 – 2 of 2) sorted by relevance
| /dports/math/py-autograd/autograd-1.3/autograd/ |
| H A D | core.py | 31 vjpmaker = primitive_vjps[fun]
36 self.vjp = vjpmaker(parent_argnums, value, args, kwargs)
43 def defvjp_argnums(fun, vjpmaker): argument
44 primitive_vjps[fun] = vjpmaker
46 def defvjp_argnum(fun, vjpmaker): argument
48 vjps = [vjpmaker(argnum, *args) for argnum in argnums]
54 vjps_dict = {argnum : translate_vjp(vjpmaker, fun, argnum)
55 for argnum, vjpmaker in zip(argnums, vjpmakers)}
289 def defvjp_unstaged(vjpmaker, argnum=0): argument
294 return vjpmaker(g, ans, vs, gvs, *args, **kwargs)
|
| /dports/math/py-jax/jax-0.2.9/jax/interpreters/ |
| H A D | ad.py | 742 def vjpmaker(*primals): function
747 defvjp_all(prim, vjpmaker)
750 def vjpmaker(*primals): function
755 defvjp_all(prim, vjpmaker)
|