Home
last modified time | relevance | path

Searched refs:_raise_custom_vjp_error_on_jvp (Results 1 – 2 of 2) sorted by relevance

/dports/math/py-jax/jax-0.2.9/jax/interpreters/
H A Dad.py679 def _raise_custom_vjp_error_on_jvp(*_, **__): function
682 custom_lin_p.def_impl(_raise_custom_vjp_error_on_jvp)
/dports/math/py-jax/jax-0.2.9/jax/
H A Dcustom_derivatives.py667 batching.primitive_batchers[ad.custom_lin_p] = ad._raise_custom_vjp_error_on_jvp