Searched defs:defvjp (Results 1 – 4 of 4) sorted by relevance
52 def defvjp(fun, *vjpmakers, **kwargs): function
422 def defvjp(self, fwd, bwd): member in custom_vjp
2630 def defvjp(fun, *vjprules): function
741 def defvjp(prim, *vjps): function